github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-18 17:31:53 +00:00
Code Issues Packages Projects Releases Wiki Activity
feat/ppo
ColossalAI/applications/ColossalChat/coati/distributed
History
pre-commit-ci[bot] 6e096362ef [pre-commit.ci] auto fixes from pre-commit.com hooks 
for more information, see https://pre-commit.ci
2025-03-07 10:43:03 +00:00
..
reward add ppo 2025-03-07 18:29:34 +08:00
__init__.py [chat] add distributed impl (#6210) 2025-02-21 15:24:23 +08:00
comm.py [chat] add distributed impl (#6210) 2025-02-21 15:24:23 +08:00
consumer.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2025-03-07 10:43:03 +00:00
grpo_consumer.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2025-03-06 08:29:59 +00:00
inference_backend.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2025-03-07 10:43:03 +00:00
launch.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2025-03-07 10:43:03 +00:00
loss.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2025-03-07 10:43:03 +00:00
ppo_consumer.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2025-03-07 10:43:03 +00:00
producer.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2025-03-07 10:43:03 +00:00
README.md [chat] add distributed impl (#6210) 2025-02-21 15:24:23 +08:00
utils.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2025-03-07 10:43:03 +00:00

README.md

Requirements

pip install cupy-cuda12x
python -m cupyx.tools.install_library --cuda 12.x --library nccl