谷歌JAX发布v0.12.0：新增MXFP4量化与代理后端，移除旧重算接口
JAX v0.12.0引入MXFP4量化以优化AMD GPU显存占用，并增加连接Pathways on Cloud的代理后端，同时移除了jax.custom_remat接口。
重要性局部证据E2 未复现写法快讯
由谷歌开发的开源机器学习框架JAX于2026年10月19日发布v0.12.0版本。该更新重点增强了底层计算效率与硬件适配能力，特别是针对大规模模型训练的显存优化。
在功能方面，JAX.nn.get_scaled_dot_general_config现在接受'mxfp4'格式（一种每32个元素共享一个缩放因子的低比特量化方案），并在AMD gfx950 GPU上实现原生执行，其他平台则通过反量化回退机制支持。此外，JAX注册了一个新的'proxy'后端，允许用户通过设置环境变量连接到IFRT代理服务器（如Pathways on Cloud），简化了云端分布式计算的配置流程。
此次更新也包含若干破坏性变更。最显著的是移除了jax.custom_remat函数，开发者需改用jax.custom_vjp.defremat来实现相同的功能。同时，JAX在x86-64架构上不再支持仅具备AVX指令集的旧款CPU，最低要求提升至Intel Haswell或更新型号（约2013年后发布，需支持AVX2和FMA）。NumPy依赖版本也相应提高，现要求NumPy 2.2或更高版本。