diff --git a/requirements_cpu.txt b/requirements_cpu.txt index 6dd46ad..1057f6e 100644 --- a/requirements_cpu.txt +++ b/requirements_cpu.txt @@ -6,4 +6,4 @@ # requirements_test.txt for test-only dependencies jax==0.3.16 -jaxlib==0.3.15 +jaxlib==0.3.16 diff --git a/requirements_ipu.txt b/requirements_ipu.txt index 8b63733..14c4d56 100644 --- a/requirements_ipu.txt +++ b/requirements_ipu.txt @@ -5,5 +5,5 @@ # requirements_cpu.txt for cpu backend configuration # requirements_test.txt for test-only dependencies jax @ https://github.com/graphcore-research/jax-experimental/releases/download/jax-v0.3.16-ipu-beta3-sdk3/jax-0.3.16%2Bipu-py3-none-any.whl -jaxlib @ https://github.com/graphcore-research/jax-experimental/releases/download/jax-v0.3.16-ipu-beta3-sdk3/jaxlib-0.3.15%2Bipu.sdk320-cp38-none-manylinux2014_x86_64.whl +jaxlib @ https://github.com/graphcore-research/jax-experimental/releases/download/jax-v0.3.16-ipu-beta3-sdk3/jaxlib-0.3.16%2Bipu.sdk320-cp38-none-manylinux2014_x86_64.whl tessellate-ipu@git+https://github.com/graphcore-research/tessellate-ipu.git@main