GitHub repository: sandbox-quantum/jax_ib
Description: Immersed Boundary implementation in JAX-CFD