Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Rollback to Jax 0.4.33 #854

Merged
merged 1 commit into from
Nov 22, 2024
Merged

Rollback to Jax 0.4.33 #854

merged 1 commit into from
Nov 22, 2024

Conversation

kelvin-zou
Copy link
Contributor

Roll back to fix a few bugs in TPU. Will roll forward with a higher version Jax later.

@kelvin-zou kelvin-zou marked this pull request as ready for review November 21, 2024 23:20
ruomingp
ruomingp approved these changes Nov 22, 2024
View reviewed changes
Copy link
Contributor

@ruomingp ruomingp left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks!

@kelvin-zou kelvin-zou added this pull request to the merge queue Nov 22, 2024
Merged via the queue into apple:main with commit d0edead Nov 22, 2024
10 checks passed
@kelvin-zou kelvin-zou deleted the rollback_jax_433 branch November 22, 2024 01:22
jiya-zhang added a commit to jesus-orozco/axlearn that referenced this pull request Nov 27, 2024
@jiya-zhang
Revert "snapshot (apple#854)" commit, which downgraded jax and orbax
573eafb 
This reverts commit d0edead.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@ruomingp ruomingp ruomingp approved these changes

@markblee markblee Awaiting requested review from markblee markblee is a code owner

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@kelvin-zou @ruomingp