Trigger docker rebuild

neel04 · May 4, 2024 · 32e05d8 · 32e05d8
1 parent dded270
commit 32e05d8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -18,6 +18,7 @@ RUN pip3 install Ipython matplotlib
 RUN pip3 install numpy pandas scipy
 
 RUN pip3 install -U -q jax[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
+RUN pip3 install -U jaxlib
 RUN pip3 install -q transformers datasets scalax tokenizers icecream wandb einops torch tqdm jaxtyping optax optuna equinox rich
 RUN pip3 install tensorboard-plugin-profile comet-ml optuna-integration plotly