Closed digicosmos86 closed 1 month ago
JAX has recently updated how the package should be installed. They have dropped the need to install jaxlib at the exact same version. We need to update our dependency list to exclude jaxlib and improve documentation on JAX with GPU support.
jaxlib
Tasks:
JAX has recently updated how the package should be installed. They have dropped the need to install
jaxlibat the exact same version. We need to update our dependency list to excludejaxliband improve documentation on JAX with GPU support.Tasks:
jaxlibfrom HSSM's dependency