Blue Space 70bd3d3d6b [feat] Wandb Timing: Add more detailed timing of gen_sequence and weights resharding (#1834)

2025-06-07 07:45:50 +08:00

..

config

[Refactor] fused kernel in forward (#1624)

2025-05-24 13:50:57 +08:00

__init__.py

[dev] feat: immigrate from yapf & pylint to ruff based on pre-commit (#1010)

2025-04-18 07:49:31 -07:00

main_prime.py

feat: move AsyncLLM ChatCompletionScheduler to separate thread (#1274)

2025-04-27 22:02:52 +08:00

prime_core_algos.py

[Lint] fix: linting errors in all files (#1280)

2025-04-27 15:24:30 -07:00

prime_dp_rm.py

[Refactor] fused kernel in forward (#1624)

2025-05-24 13:50:57 +08:00

prime_fsdp_workers.py

[logging] misc: update PR template and fix lint (#1806)

2025-06-04 07:53:12 +08:00

prime_ray_trainer.py

[feat] Wandb Timing: Add more detailed timing of gen_sequence and weights resharding (#1834)

2025-06-07 07:45:50 +08:00

run_prime_qwen_code.sh

[recipe] prime: Code example for PRIME (#1714)

2025-06-02 19:08:11 -07:00

run_prime_qwen.sh

fix: remove deprecated remove_previous_ckpt key in prime_ray_trainer.py (#1254)

2025-04-25 18:12:18 +08:00