推销JAX在复杂计算中的强大能力

Eric Volkmann

介绍 gyaradax 🐉：一个用于局部通量管陀螺动力学的JAX求解器，采用自定义CUDA内核加速。整个代码由@ggalletti_和我在一个月内通过vibecode完成。已针对GKW（仅CPU的Fortran代码）验证，加速10倍。详情和代码见回复。https://t.co/22PrHjItR5

打开互动版