License.txt
README.md
pyproject.toml
memzero_train.egg-info/PKG-INFO
memzero_train.egg-info/SOURCES.txt
memzero_train.egg-info/dependency_links.txt
memzero_train.egg-info/requires.txt
memzero_train.egg-info/top_level.txt
mztrain/__init__.py
mztrain/core/__init__.py
mztrain/core/autograd.py
mztrain/core/memory.py
mztrain/core/ops.py
mztrain/core/tensor.py
mztrain/data/__init__.py
mztrain/data/stream.py
mztrain/data/synthetic.py
mztrain/models/__init__.py
mztrain/models/cnn.py
mztrain/models/minitf.py
mztrain/models/mlp.py
mztrain/optim/__init__.py
mztrain/optim/adam8bit.py
mztrain/optim/sgd.py
mztrain/train/__init__.py
mztrain/train/parallel.py
mztrain/train/trainer.py
mztrain/utils/__init__.py
mztrain/utils/system.py
tests/test_adam8bit.py
tests/test_autograd.py
tests/test_stream.py
tests/test_tensor.py
tests/test_trainer.py