支持 gemma4 系列的 Megatron 训练

### Checklist / 检查清单

- [x] I have searched existing issues, and this is a new feature request. / 我已经搜索过现有的 issues，确认这是一个新的 Feature Request。

### Feature Request Description / Feature Request 描述

如题，辛苦作者大大~

### Pull Request / Pull Request 信息

_No response_