`del self._old_env` so that you can use `with global_mesh` multiple times. This was Matt's idea.
`del self._old_env` so that you can use `with global_mesh` multiple times. This was Matt's idea.

Co-authored-by: Matthew Johnson <e68678f07030b810d1440f7f1c1c423f5afe56fe@google.com>
PiperOrigin-RevId: 429229193

google/jax,google/jax,google/jax,google/jax