PR #7601 Update JAX version to fix TPU CI

Update JAX version to fix TPU CI #7601

will-cromar merged 1 commit into master from wcromar/jax-4-31

Update JAX version to fix TPU CI

827d91de

will-cromar added tpuci

will-cromar requested a review from

JackCaoG 1 year ago

JackCaoG approved these changes on 2024-07-01

will-cromar merged 593dce8b into master 1 year ago

Reviewers

JackCaoG

Assignees

No one assigned

Labels

None yet

Milestone

No milestone