Liam-Sturge opened a new pull request, #15421: URL: https://github.com/apache/tvm/pull/15421
The Flax dependency `Orbax v0.1.8` has enforced the deprecation of installing Orbax as a standalone package. Flax v0.6.8 (which is currently pinned in the script `docker/install/ubuntu_install_jax.sh`) attempts to install Orbax as a standalone package and raises an error about doing so. Going forward, the package `orbax-checkpoint` should be installed instead. Flax v0.6.8 does not recognise this and attempts to install Orbax instead of orbax-checkpoint and the installation fails. In order to resolve Jax installation issues, bumping the version of Flax to be at least 0.6.9, which resolves the problem. Flax >= 0.6.9 does not pin the version of orbax-checkpoint that it installs and the latest version requires Jax >= 0.4.9 to be installed so the two must be updated together. Issue: [15420](https://github.com/apache/tvm/issues/15420) raised that details the error which occurs and steps to reproduce. cc @leandron @Mousius @lhutton1 @ekalda @areusch @konturn @tqchen -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
