From 1eeb7297e698fd0e54e63be23dc469cf71898a5f Mon Sep 17 00:00:00 2001 From: Hatem Helal Date: Thu, 9 Nov 2023 14:48:08 +0000 Subject: [PATCH] try fixing req --- requirements_cpu.txt | 2 +- requirements_ipu.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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