LICENSE
README.md
pyproject.toml
stream_transformer/__init__.py
stream_transformer/cli.py
stream_transformer/engine.py
stream_transformer/kv_cache.py
stream_transformer/prefetcher.py
stream_transformer/sharder.py
stream_transformer/trainer.py
stream_transformer/utils.py
stream_transformer.egg-info/PKG-INFO
stream_transformer.egg-info/SOURCES.txt
stream_transformer.egg-info/dependency_links.txt
stream_transformer.egg-info/entry_points.txt
stream_transformer.egg-info/requires.txt
stream_transformer.egg-info/top_level.txt
stream_transformer/models/__init__.py
stream_transformer/models/llama.py
tests/test_engine.py