some edits

nvfp4gguf/__init__.py

@@ -9,6 +9,10 @@ from .quant import (
     DEFAULT_EARLY_STOP_STALL,
     DEFAULT_SCALE_OPTIMIZATION,
     DEFAULT_SCALE_REFINEMENT_ROUNDS,
+    FP4_LEVELS,
+    FP4_LEVELS_SORTED,
+    FP4_LEVEL_MAP,
+    FP4_TABLE,
     QUANT_VERSION,
     ShapeView,
     TensorQuantizer,
@@ -25,16 +29,21 @@ from .quant import (
     must_bake_nvfp4_scale,
     mtp_layer_tensor,
     bake_packed_nvfp4,
+    optimize_block_scales,
+    optimize_block_scales_torch,
     pack_nvfp4,
     preserve_scale_tensor,
     quant_profile,
     quant_settings,
     quantize_and_pack,
+    refine_codes_for_scale,
+    refine_codes_for_scale_torch,
     report_reconstruction,
     require_quant_settings,
     resolve_optimizer,
     scale_bake_notice,
     ue4m3_bits,
+    ue4m3_bits_torch,
 )
 
 __all__ = [
@@ -46,6 +55,10 @@ __all__ = [
     "DEFAULT_EARLY_STOP_STALL",
     "DEFAULT_SCALE_OPTIMIZATION",
     "DEFAULT_SCALE_REFINEMENT_ROUNDS",
+    "FP4_LEVELS",
+    "FP4_LEVELS_SORTED",
+    "FP4_LEVEL_MAP",
+    "FP4_TABLE",
     "QUANT_VERSION",
     "ShapeView",
     "TensorQuantizer",
@@ -62,14 +75,19 @@ __all__ = [
     "must_bake_nvfp4_scale",
     "mtp_layer_tensor",
     "bake_packed_nvfp4",
+    "optimize_block_scales",
+    "optimize_block_scales_torch",
     "pack_nvfp4",
     "preserve_scale_tensor",
     "quant_profile",
     "quant_settings",
     "quantize_and_pack",
+    "refine_codes_for_scale",
+    "refine_codes_for_scale_torch",
     "report_reconstruction",
     "require_quant_settings",
     "resolve_optimizer",
     "scale_bake_notice",
     "ue4m3_bits",
+    "ue4m3_bits_torch",
 ]