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 2021-10-28 16:21:23 +00:00			`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) 2022-01-28 08:59:53 +00:00			`packaging`
			`pre-commit`
[log] better logging display with rich (#426) * better logger using rich * remove deepspeed in zero requirements 2022-03-16 01:51:15 +00:00			`rich`
[cli] add missing requirement (#805) 2022-04-19 05:56:59 +00:00			`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 2022-04-24 05:26:26 +00:00			`fabric`
[setup] supported conda-installed torch (#2048) * [setup] supported conda-installed torch * polish code 2022-11-30 08:45:15 +00:00			`contexttimer`
[buider] use builder() for cpu adam and fused optim in setup.py (#2187) 2022-12-23 08:05:13 +00:00			`ninja`
fix torch 2.0 compatibility (#3346) 2023-03-30 07:25:24 +00:00			`torch>=1.11`
[checkpoint] refactored the API and added safetensors support (#3427) * [checkpoint] refactored the API and added safetensors support * polish code 2023-04-04 07:23:01 +00:00			`safetensors`
[kernel] updated unittests for coloattention (#4389) Updated coloattention tests of checking outputs and gradients 2023-08-09 06:24:45 +00:00			`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 2023-08-04 09:42:07 +00:00			`einops`