mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-04 02:26:51 +00:00

Files

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

add ppo

2025-03-07 18:29:34 +08:00

__init__.py

2025-02-21 15:24:23 +08:00

comm.py

2025-02-21 15:24:23 +08:00

consumer.py

2025-03-07 10:43:03 +00:00

grpo_consumer.py

2025-03-06 08:29:59 +00:00

inference_backend.py

2025-03-07 10:43:03 +00:00

launch.py

2025-03-07 10:43:03 +00:00

loss.py

2025-03-07 10:43:03 +00:00

ppo_consumer.py

2025-03-07 10:43:03 +00:00

producer.py

2025-03-07 10:43:03 +00:00

README.md

2025-02-21 15:24:23 +08:00

utils.py

2025-03-07 10:43:03 +00:00

Requirements

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