jax_tpu_docker
★ 1
updated 3y ago
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.
No plain-English explanation yet — one is being written right now. Check back in a minute.