jax

627debc7 - Create a `null_mesh_context` internal context manager to handle null contexts properly.