diff --git a/images/tpu/jax/Dockerfile.x86_64 b/images/tpu/jax/Dockerfile.x86_64
new file mode 100644
index 000000000..d1e088b1c
--- /dev/null
+++ b/images/tpu/jax/Dockerfile.x86_64
@@ -0,0 +1,4 @@
+FROM python:3.8
+
+RUN pip install --upgrade jax
+RUN pip install jax[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html