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
JAX-Fluids: JAX写的可微流体动力学包,运行在 CPU/GPU/TPU 上,支持自动微分对数值模型进行端到端优化JAX-Fluids: JAX写的可微流体动力学包,运行在 CPU/GPU/TPU 上,支持自动微分对数值模型进行端到端优化JAX-Fluids: JAX写的可微流体动力学包,运行在 CPU/GPU/TPU 上,支持自动微分对数值模型进行端到端优化