JAX发布新版本：提供逐函数重计算控制API

JAX是Google的可变换数值计算库，广泛用于大规模机器学习研究。本次正式版提供逐函数重计算控制API：开发者可对个别函数指定重计算（gradient checkpointing）策略，而非仅做全局选择，实质扩展计算与内存的控制粒度。在显存受限场景下做大规模训练与推理的团队，可按算子粒度在重算开销与显存占用间取得更细的平衡；API具体语义与适用范围以官方版本说明为准。