Immersed Boundary implementation in JAX-CFD - View it on GitHub