You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/pipeline
Wang Binluo eea37da6fa
[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)
3 months ago
..
schedule [fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016) 3 months ago
__init__.py [misc] update pre-commit and run all files (#4752) 1 year ago
p2p.py [FP8] rebase main (#5963) 4 months ago
stage_manager.py [Feature] optimize PP overlap (#5735) 5 months ago