fix: keep Adam moments in FP32 for BF16 parameters

[JYMiracle305]

修改背景

Adam 的一阶动量和二阶动量会在训练过程中持续累积。原实现让 m、v 跟随模型参数的数据类型，BF16 参数会因此使用 BF16 optimizer state，容易损失较小的梯度变化。

主要修改

• Adam 一阶动量 m 和二阶动量 v 统一使用 FP32 保存和计算
• CUDA Adam kernel 分别处理梯度类型、参数类型和 FP32 optimizer state
• 同时覆盖普通参数列表和 named-parameter 两种 Adam 构造路径
• 增加 BF16 参数使用 FP32 Adam state 的创建和更新测试

该 PR 位于 checkpoint stack 中，依赖 #195，后续 PR 为 #196。