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.

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

Pins jax==0.4.23 (xla matched), flax==0.7.5, te==1.2.1 #493

Draft
wants to merge 2 commits into
base: archive-24.01-devel
Choose a base branch
from

Conversation

terrykong
Copy link
Contributor

@terrykong terrykong commented Jan 23, 2024

This PR should not be merged if the t5x/pax tests fail and may require us unpinning or bumping the core libraries or downgrading pax/t5x.

To review:

@terrykong terrykong marked this pull request as ready for review January 23, 2024 17:28
@terrykong terrykong marked this pull request as draft January 23, 2024 17:28
@terrykong
Copy link
Contributor Author

@DwarKapex So flax==0.7.5 is not new enough for the patches in version control from when we cut this release branch. I'm going to revert it to see if that resolves the build issues for rosetta

@terrykong
Copy link
Contributor Author

Looks like JAX ARM builds aren't succeeding, possibly due to the patch @DwarKapex . Can you look into this and decide how you'd like to proceed?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant