ColossalAI/requirements/requirements.txt

numpy
tqdm
psutil
packaging
pre-commit
rich
click
fabric
contexttimer
ninja
torch>=1.11
safetensors
sortedcontainers
einops
Migrated project 3 years ago			`numpy`
			`tqdm`
			`psutil`
Update GitHub action and pre-commit settings (#196) * Update GitHub action and pre-commit settings * Update GitHub action and pre-commit settings (#198) 3 years ago			`packaging`
			`pre-commit`
[log] better logging display with rich (#426) * better logger using rich * remove deepspeed in zero requirements 3 years ago			`rich`
[cli] add missing requirement (#805) 3 years ago			`click`
[CLI] refactored the launch CLI and fixed bugs in multi-node launching (#844) * [cli] fixed multi-node job launching * [cli] fixed a bug in version comparison * [cli] support launching with env var * [cli] fixed multi-node job launching * [cli] fixed a bug in version comparison * [cli] support launching with env var * added docstring * [cli] added extra launch arguments * [cli] added default launch rdzv args * [cli] fixed version comparison * [cli] added docstring examples and requierment * polish docstring * polish code * polish code 3 years ago			`fabric`
[setup] supported conda-installed torch (#2048) * [setup] supported conda-installed torch * polish code 2 years ago			`contexttimer`
[buider] use builder() for cpu adam and fused optim in setup.py (#2187) 2 years ago			`ninja`
fix torch 2.0 compatibility (#3346) 2 years ago			`torch>=1.11`
[checkpoint] refactored the API and added safetensors support (#3427) * [checkpoint] refactored the API and added safetensors support * polish code 2 years ago			`safetensors`
[elixir] add elixir and its unit tests (#3835) * [elixir] add elixir * [elixir] add unit tests * remove useless code * fix python 3.8 issue * fix typo * add test skip * add docstrings * add docstrings * add readme * fix typo 2 years ago			`sortedcontainers`
			`einops`