Remove support from static_argnums, as it is not a hashable type (and now causes an error). More details: https://github.com/google/jax/pull/4717
Remove support from static_argnums, as it is not a hashable type (and now causes an error). More details: https://github.com/google/jax/pull/4717

PiperOrigin-RevId: 350190708

google/dopamine,google/dopamine,google/dopamine