Metadata-Version: 2.1
Name: colossalai
Version: 0.0.2-gitc789114680-dtk2210
Summary: An integrated large-scale model training system with efficient parallelization techniques
Home-page: UNKNOWN
License: UNKNOWN
Platform: UNKNOWN
Requires-Dist: torch (>=1.8)
Requires-Dist: torchvision (>=0.9)
Requires-Dist: numpy
Requires-Dist: tqdm
Requires-Dist: psutil
Requires-Dist: tensorboard
Requires-Dist: packaging
Requires-Dist: pre-commit
Provides-Extra: zero
Requires-Dist: deepspeed ; extra == 'zero'

UNKNOWN


