mirror of
https://github.com/ROCm/jax.git
synced 2025-04-17 20:36:05 +00:00

-- d39bdefb33a19e407c352df27fb04127f4fe8a1d by Peter Hawkins <phawkins@google.com>: Migrate more tests from jtu.cases_from_list to jtu.sample_product. COPYBARA_INTEGRATE_REVIEW=https://github.com/google/jax/pull/12717 from hawkinsp:sampletest d39bdefb33a19e407c352df27fb04127f4fe8a1d PiperOrigin-RevId: 480136538