JAX-Fluids: JAX写的可微流体动力学包,运行在 CPU/GPU/TPU 上,支持自动微分对数值模型进行端到端优化 爱可可-爱生活 2022-11-07 21:58:40 【JAX-Fluids: JAX写的可微流体动力学包,运行在 CPU/GPU/TPU 上,支持自动微分对数值模型进行端到端优化】’JAX-Fluids: A Differentiable Fluid Dynamics Package' GitHub: github.com/tumaer/JAXFLUIDS 开源JAX