Their release procedure is to tag a preliminary release on GitHub and start testing, but the the actual release comes when the wheels are pushed to PyPI.
In v0.3.14, there was a substantial revision between the GitHub release after 7 days. This requires a rebuild on our end and we should mark our build as broken as it won't truly reflect the "release" version.
To avoid this issue in the future, we should only get our sources from PyPI.
Solution to issue cannot be found in the documentation.
Issue
The JAX team updated their tags post-release due to issues with NCCL. See more details here: https://github.com/google/jax/issues/11265.
Their release procedure is to tag a preliminary release on GitHub and start testing, but the the actual release comes when the wheels are pushed to PyPI.
In v0.3.14, there was a substantial revision between the GitHub release after 7 days. This requires a rebuild on our end and we should mark our build as broken as it won't truly reflect the "release" version.
To avoid this issue in the future, we should only get our sources from PyPI.
Installed packages
Environment info