AeroJAX: JAX-native CFD, differentiable end-to-end. ~560 FPS at 128×128 on CPU [P]
I have been building a JAX based CFD framework for differentiable Navier Stokes simulation inside ML loops such as inverse design and learned closures. The goal is to keep the full solver stack differentiable so it can sit inside optimisation and learn…