"""Project-local native NVFP4 experiments for Fish Audio S2-Pro."""

from .checkpoint import load_mixed_nvfp4_checkpoint
from .server import launch_mixed_nvfp4_thread_safe_queue

from .modules import (
    NVFP4FeedForward,
    NVFP4Linear,
    NVFP4TransformerBlock,
    apply_nvfp4_rounding_checkpoint,
    convert_s2_pro_nvfp4,
    load_direct_w4a4_m1_extension,
    load_rmsnorm_nvfp4_m1_extension,
    load_smallm_nvfp4_extension,
    load_silu_product_nvfp4_m1_extension,
    quantize_nvfp4,
)

__all__ = [
    "NVFP4FeedForward",
    "NVFP4Linear",
    "NVFP4TransformerBlock",
    "apply_nvfp4_rounding_checkpoint",
    "convert_s2_pro_nvfp4",
    "load_mixed_nvfp4_checkpoint",
    "launch_mixed_nvfp4_thread_safe_queue",
    "load_direct_w4a4_m1_extension",
    "load_rmsnorm_nvfp4_m1_extension",
    "load_smallm_nvfp4_extension",
    "load_silu_product_nvfp4_m1_extension",
    "quantize_nvfp4",
]