Add jax, the vetter np version.

David Thrower · David Thrower · commit b93607e65905 · 2026-01-20T14:05:42.000-05:00
diff --git a/Dockerfile b/Dockerfile
@@ -22,5 +22,6 @@ RUN echo "##### END OF LIST OF EXISTING PYTHON PACKAGES #####"
 ## / debug
 
 RUN pip install --upgrade pip
+# RUN pip install "numpy==1.26.4" --no-deps
 RUN pip install --ignore-installed blinker --ignore-installed ml_dtypes -r docker-requirements.txt
 RUN python train_a_generative_llm.py
diff --git a/docker-requirements.txt b/docker-requirements.txt
@@ -1,8 +1,8 @@
-# jax==0.5.3
-# jaxlib==0.5.3
+jax[cuda12_pip]
+jaxlib==0.8.0
 pendulum==3.0.0
 # tensorflow==2.20.0
-# numpy==2.3.5
+numpy==2.3.5
 pandas==2.3.3
 # pyvis==0.3.2
 # plotly==5.20.0
