LICENSE
README.md
pyproject.toml
halo/__init__.py
halo/hub.py
halo/attention/__init__.py
halo/attention/global_attention.py
halo/attention/graph.py
halo/attention/halo_attention.py
halo/core/__init__.py
halo/core/config.py
halo/core/device.py
halo/core/logging.py
halo/datasets/__init__.py
halo/datasets/jsonl.py
halo/datasets/streaming.py
halo/datasets/synthetic.py
halo/datasets/text.py
halo/generation/__init__.py
halo/generation/samplers.py
halo/models/__init__.py
halo/models/baseline_model.py
halo/models/halo_model.py
halo/nn/__init__.py
halo/nn/feed_forward.py
halo/nn/halo_block.py
halo/nn/rope.py
halo/tokenizers/__init__.py
halo/tokenizers/base.py
halo/tokenizers/char.py
halo/tokenizers/sentencepiece.py
halo/tokenizers/word.py
halo/training/__init__.py
halo/training/trainer.py
halo/utils/__init__.py
halo/utils/benchmarks.py
halo/utils/metrics.py
halo/utils/random.py
pyhalos.egg-info/PKG-INFO
pyhalos.egg-info/SOURCES.txt
pyhalos.egg-info/dependency_links.txt
pyhalos.egg-info/requires.txt
pyhalos.egg-info/top_level.txt
tests/test_attention.py
tests/test_checkpoint.py
tests/test_config.py
tests/test_generation.py
tests/test_gradients.py
tests/test_graph.py
tests/test_memory.py
tests/test_model.py
tests/test_shapes.py
tests/test_tokenizers.py
tests/test_training.py