LICENSE
MANIFEST.in
NOTICE
README.md
build.bat
pyproject.toml
setup.py
tests/test_distributed.py
tests/test_features.py
tests/test_huggingface.py
tests/test_offload.py
tests/test_optimizer.py
tests/test_paging.py
trainstream/__init__.py
trainstream/_build.py
trainstream/hooks.py
trainstream/losses.py
trainstream/optimizer.py
trainstream/paging.py
trainstream/profiler.py
trainstream/wrapper.py
trainstream.egg-info/PKG-INFO
trainstream.egg-info/SOURCES.txt
trainstream.egg-info/dependency_links.txt
trainstream.egg-info/not-zip-safe
trainstream.egg-info/requires.txt
trainstream.egg-info/top_level.txt
trainstream/csrc/bindings.cpp
trainstream/csrc/kernels.h
trainstream/csrc/memory_pinner.cpp
trainstream/csrc/offload_adamw.cpp
trainstream/csrc/offload_adamw.h
trainstream/csrc/offload_adamw_kernel.cu
trainstream/csrc/trainstream.h
trainstream/integrations/__init__.py
trainstream/integrations/huggingface.py