jax

fa1538d2 - Enter into `eqn.ctx` in checkify's jaxpr interpreter so that the correct context mesh is used.