# GPU加速JAX

四个大模型把Kessler暖雨微物理的Fortran代码改写成GPU上的JAX，输出与Fortran参考几乎一致，但GPU性能相差很大，最快一版在缓存GPU执行下比串行Fortran快约357倍。

四个大模型把Kessler暖雨微物理的Fortran代码改写成GPU上的JAX，输出与Fortran参考几乎一致，但GPU性能相差很大，最快一版在缓存GPU执行下比串行Fortran快约357倍。