LICENSE
README.md
pyproject.toml
setup.py
axiom1/README.md
axiom1/__init__.py
axiom1.egg-info/PKG-INFO
axiom1.egg-info/SOURCES.txt
axiom1.egg-info/dependency_links.txt
axiom1.egg-info/requires.txt
axiom1.egg-info/top_level.txt
axiom1/evals/__init__.py
axiom1/evals/loop_stats.py
axiom1/evals/perplexity.py
axiom1/evals/sanity_check.py
axiom1/inference/__init__.py
axiom1/inference/generate.py
axiom1/inference/kv_cache.py
axiom1/inference/memory.py
axiom1/inference/offload.py
axiom1/inference/quantize.py
axiom1/model/__init__.py
axiom1/model/block.py
axiom1/model/components.py
axiom1/model/config.py
axiom1/model/model.py
axiom1/tokenizer/__init__.py
axiom1/tokenizer/build_tokenizer.py
axiom1/tokenizer/tokenizer.py
axiom1/training/__init__.py
axiom1/training/data.py
axiom1/training/loss.py
axiom1/training/optimizer.py
axiom1/training/phases.py
axiom1/training/train.py
test/test_components.py
test/test_config.py
test/test_e2e.py
test/test_inference.py
test/test_model.py
test/test_model_chat.py
test/test_tokenizer.py
test/test_training.py