diff --git a/bit_jax/requirements.txt b/bit_jax/requirements.txt index 041a271..be1111e 100644 --- a/bit_jax/requirements.txt +++ b/bit_jax/requirements.txt @@ -1,5 +1,5 @@ flax==0.1.0 -numpy==1.18.4 +numpy==1.22.0 tensorflow-cpu==2.2.0 tensorflow-datasets==3.1.0 tensorflow-probability==0.9.0