LICENSE
MANIFEST.in
NOTICE
README.md
build.bat
pyproject.toml
setup.py
csrc/bindings.cpp
csrc/kernels.h
csrc/memory_pinner.cpp
csrc/offload_adamw.cpp
csrc/offload_adamw.h
csrc/offload_adamw_kernel.cu
csrc/trainstream.h
tests/test_huggingface.py
tests/test_offload.py
tests/test_optimizer.py
trainstream/__init__.py
trainstream/hooks.py
trainstream/optimizer.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/integrations/__init__.py
trainstream/integrations/huggingface.py