Use `contextlib.nullcontext` instead of `trivial_ctx`
I removed `trivial_ctx` from the public `jax.interpreters.partial_eval`
submodule without going through a deprecation cycle, because it is highly
unlikely anyone is using it.
PiperOrigin-RevId: 744645764
