LICENSE
README.md
pyproject.toml
axiomlm/__init__.py
axiomlm/train.py
axiomlm.egg-info/PKG-INFO
axiomlm.egg-info/SOURCES.txt
axiomlm.egg-info/dependency_links.txt
axiomlm.egg-info/entry_points.txt
axiomlm.egg-info/requires.txt
axiomlm.egg-info/top_level.txt
axiomlm/dengine/__init__.py
axiomlm/dengine/dataloader.py
axiomlm/engine/__init__.py
axiomlm/engine/export.py
axiomlm/engine/inference.py
axiomlm/engine/paged_cache.py
axiomlm/kernels/__init__.py
axiomlm/kernels/benchmark_kernels.py
axiomlm/kernels/build_kernels.py
axiomlm/kernels/ops.py
axiomlm/kernels/triton_kernels.py
axiomlm/models/__init__.py
axiomlm/models/attention.py
axiomlm/models/modules.py
axiomlm/models/transformer.py
axiomlm/optim/__init__.py
axiomlm/optim/muon.py
axiomlm/optim/schedule.py
axiomlm/telemetry/__init__.py
axiomlm/telemetry/profiler.py
tests/test_all.py
tests/test_kernels.py