README.md
pyproject.toml
coat/__init__.py
coat/fp8_trainer.py
coat/activation/__init__.py
coat/activation/utils.py
coat/activation/fake_quantization/FloatPointQuantizeTorch.py
coat/activation/fake_quantization/FloatPointQuantizeTriton.py
coat/activation/fake_quantization/__init__.py
coat/activation/fake_quantization/quantize_function.py
coat/activation/fake_quantization/utils.py
coat/activation/models/_fp8_quantization_config.py
coat/activation/models/_fp8_weightcache.py
coat/activation/models/_fp8manager.py
coat/activation/models/coat_llama.py
coat/activation/models/coat_llama_convert_from_hf.py
coat/activation/models/coat_olmo.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_pertensor.py
coat/activation/real_quantization/_quantize_pertensor_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/optimizer/__init__.py
coat/optimizer/fp8_adamw.py
coat/optimizer/kernels/setup.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_utils.py