README.rst
setup.py
model_wrapper/__init__.py
model_wrapper/activations.py
model_wrapper/dataset.py
model_wrapper/log_utils.py
model_wrapper/norm.py
model_wrapper/utils.py
model_wrapper.egg-info/PKG-INFO
model_wrapper.egg-info/SOURCES.txt
model_wrapper.egg-info/dependency_links.txt
model_wrapper.egg-info/not-zip-safe
model_wrapper.egg-info/requires.txt
model_wrapper.egg-info/top_level.txt
model_wrapper/model/__init__.py
model_wrapper/model/_attention.py
model_wrapper/model/embedding.py
model_wrapper/model/layer.py
model_wrapper/model/rotary_emb.py
model_wrapper/training/__init__.py
model_wrapper/training/_support.py
model_wrapper/training/_trainer.py