Files
ColossalAI/requirements/requirements.txt
Hongxin Liu 3788fefc7a [zero] support multiple (partial) backward passes (#5596)
* [zero] support multiple (partial) backward passes

* [misc] update requirements
2024-04-16 17:49:21 +08:00

19 lines
151 B
Plaintext

numpy
tqdm
psutil
packaging
pre-commit
rich
click
fabric
contexttimer
ninja
torch>=2.1.0
safetensors
einops
pydantic
ray
sentencepiece
google
protobuf