V0.0.5 Release Plan

[yzygitzh]

Release Manager

@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) - @abuccts
2. • Tests - Fix unit tests caused by torch.load API change (Tests - Fix unit tests caused by torch.load API change #59) - @abuccts
3. • Tests - Recover single-session tests (Tests - Recover single-session tests #64) - @yzygitzh
4. • Tests - Adjust enabled and disabled tests (Tests - Adjust enabled and disabled tests #69) - @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) - @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) - @qghuxmu