ColossalAI/requirements/requirements.txt

numpy
tqdm
psutil
packaging
pre-commit
rich
click
fabric
contexttimer
ninja
torch>=1.11
safetensors
flash_attn>=2.0
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`
[kernel] updated unittests for coloattention (#4389) Updated coloattention tests of checking outputs and gradients 1 year ago			`flash_attn>=2.0`
[doc] add Series A Funding and NeurIPS news (#4377) * [doc] add Series A Funding and NeurIPS news * [kernal] fix mha kernal * [CI] skip moe * [CI] fix requirements 1 year ago			`einops`