V0.0.5 Release Plan

[yzygitzh]

Release Manager

Ziyue Yang (@yzygitzh)

Endgame

• Code freeze: July. 7, 2025
• Bug Bash date: July. 8, 2025
• Release date: July. 11, 2025

CI/CD

1. • Tests - Fix failures in tests/unit_tests/test_checkpointing.py (Tests - Fix failures in tests/unit_tests/test_checkpointing.py #51) - Yifan Xiong (@abuccts)
2. • Tests - Fix unit tests caused by torch.load API change (Tests - Fix unit tests caused by torch.load API change #59) - Yifan Xiong (@abuccts)
3. • Tests - Recover single-session tests (Tests - Recover single-session tests #64) - Ziyue Yang (@yzygitzh)
4. • Tests - Adjust enabled and disabled tests (Tests - Adjust enabled and disabled tests #69) - Ziyue Yang (@yzygitzh)

Framework Features

MoE

1. • Algorithm - Fix all-reduce for global-batch load balancing loss (Algorithm - Fix all-reduce for global-batch load balancing loss #68) - Qingguo Hu (@qghuxmu)

TODO

MoE

1. • Algorithm - Add top-1 load balancing loss for MoE routing (Algorithm - Add top-1 load balancing loss for MoE routing #67) - Qingguo Hu (@qghuxmu)