jax

88d6a8fc - PR #28391: Add version guard in autodidax test.

PR #28391: Add version guard in autodidax test. Imported from GitHub PR https://github.com/jax-ml/jax/pull/28391 Copybara import of the project: -- 8f85229ba059f301bfb0640ba9a304dab7a345ca by Peter Hawkins <phawkins@google.com>: Add version guard in autodidax test. Merging this change closes #28391 COPYBARA_INTEGRATE_REVIEW=https://github.com/jax-ml/jax/pull/28391 from hawkinsp:add 8f85229ba059f301bfb0640ba9a304dab7a345ca PiperOrigin-RevId: 752764010