LICENSE
README.md
pyproject.toml
coat/__init__.py
coat/fp8_trainer.py
coat/activation/__init__.py
coat/activation/utils.py
coat/activation/deepseek/fp8linear.py
coat/activation/deepseek/linear.py
coat/activation/fake_quantization/FloatPointQuantizeTorch.py
coat/activation/fake_quantization/FloatPointQuantizeTriton.py
coat/activation/fake_quantization/quantize_function.py
coat/activation/fake_quantization/utils.py
coat/activation/liger/cross_entropy.py
coat/activation/liger/utils.py
coat/activation/real_quantization/__init__.py
coat/activation/real_quantization/_dequantize.py
coat/activation/real_quantization/_division.py
coat/activation/real_quantization/_division_transpose.py
coat/activation/real_quantization/_memory_io.py
coat/activation/real_quantization/_quantize.py
coat/activation/real_quantization/_quantize_perblock.py
coat/activation/real_quantization/_quantize_perblock_transpose.py
coat/activation/real_quantization/_quantize_perblock_transpose_pergroup.py
coat/activation/real_quantization/_quantize_pertensor.py
coat/activation/real_quantization/_quantize_pertensor_transpose.py
coat/activation/real_quantization/_quantize_transpose.py
coat/activation/real_quantization/_transpose.py
coat/activation/real_quantization/add_bwd.py
coat/activation/real_quantization/add_fwd.py
coat/activation/real_quantization/common.py
coat/activation/real_quantization/fp8linear.py
coat/activation/real_quantization/func_layernorm_noparam.py
coat/activation/real_quantization/func_layernorm_param.py
coat/activation/real_quantization/func_quantize.py
coat/activation/real_quantization/func_rmsnorm.py
coat/activation/real_quantization/gelu_bwd.py
coat/activation/real_quantization/gelu_bwd_legacy.py
coat/activation/real_quantization/gelu_fwd.py
coat/activation/real_quantization/linear.py
coat/activation/real_quantization/mul_bwd.py
coat/activation/real_quantization/mul_bwd_legacy.py
coat/activation/real_quantization/mul_fwd.py
coat/activation/real_quantization/silu_bwd.py
coat/activation/real_quantization/silu_bwd_legacy.py
coat/activation/real_quantization/silu_fwd.py
coat/models/__init__.py
coat/models/coat_llama.py
coat/models/coat_llama_convert_from_hf.py
coat/models/coat_olmo.py
coat/optimizer/fp8_adamw.py
coat/optimizer/kernels/bindings.cpp
coat/optimizer/kernels/fp8_adamw_cuda.cpp
coat/optimizer/kernels/fp8_adamw_cuda_kernel.cu
coat/optimizer/kernels/fp8_adamw_expand_cuda.cpp
coat/optimizer/kernels/fp8_adamw_expand_cuda_kernel.cu
coat/optimizer/kernels/setup.py
coat/utils/_fp8_quantization_config.py
coat/utils/_fp8_weightcache.py
coat/utils/_fp8manager.py
coat/utils/replacer.py
fp8_coat.egg-info/PKG-INFO
fp8_coat.egg-info/SOURCES.txt
fp8_coat.egg-info/dependency_links.txt
fp8_coat.egg-info/requires.txt
fp8_coat.egg-info/top_level.txt
tests/test_coat_llamalayer.py
tests/test_fp8_activations.py
tests/test_fp8_adamw.py
tests/test_fp8_adamw_expand.py
tests/test_fp8_linear.py
tests/test_fp8_norm.py
tests/test_fp8_quantize.py
tests/test_mx_nv_quantize.py
tests/test_utils.py