Jax TPU VM Docker
Run Docker Container
sudo docker run --net=host --privileged -it --entrypoint=/bin/bash orenleung/jax_tpu:latest
Build Docker Image
sudo docker build -t orenleung/jax_tpu .
Provides a Docker image for running JAX on TPU VMs. This repository contains the necessary files to build the Docker image and run JAX on TPUs with ease, making it easy for researchers to take advantage of the power of TPUs for their machine learning projects.
sudo docker run --net=host --privileged -it --entrypoint=/bin/bash orenleung/jax_tpu:latest
sudo docker build -t orenleung/jax_tpu .
Provides a Docker image for running JAX on TPU VMs. This repository contains the necessary files to build the Docker image and run JAX on TPUs with ease, making it easy for researchers to take advantage of the power of TPUs for their machine learning projects.