README.md
pyproject.toml
src/ultra_fused/__init__.py
src/ultra_fused/config.py
src/ultra_fused/kernels/__init__.py
src/ultra_fused/kernels/fused_sdla_kernel.py
src/ultra_fused/kernels/triton_kernels.py
src/ultra_fused/layers/__init__.py
src/ultra_fused/layers/mla_baseline.py
src/ultra_fused/layers/parallel_block.py
src/ultra_fused/layers/quant_linear.py
src/ultra_fused/layers/sdla_attention.py
src/ultra_fused/model/__init__.py
src/ultra_fused/model/transformer.py
src/ultra_fused/utils/__init__.py
src/ultra_fused/utils/mx_utils.py
src/ultra_fused/utils/yarn_rope.py
src/ultra_fused_transformer.egg-info/PKG-INFO
src/ultra_fused_transformer.egg-info/SOURCES.txt
src/ultra_fused_transformer.egg-info/dependency_links.txt
src/ultra_fused_transformer.egg-info/requires.txt
src/ultra_fused_transformer.egg-info/top_level.txt
tests/test_import.py