Following the install instructions in the documentation, I was unable to install everything with a python3.8 environment due to the jax dependency (failure to solve environment). However, creating the virtual environment with python 3.9 allowed me to install all packages and dependencies.
Following the install instructions in the documentation, I was unable to install everything with a python3.8 environment due to the
jax
dependency (failure to solve environment). However, creating the virtual environment with python 3.9 allowed me to install all packages and dependencies.It would be good to update the documentation to reflect this.