jax_xshells rotating convection