April 30, 2026

MUHAMMAD GHIFARY

Setup Cloud TPU Machine

Install Python 3.12

sudo add-apt-repository ppa:deadsnakes/ppa -y
sudo apt update
sudo apt install python3.12 python3.12-venv python3.12-dev -y

Create Virtual Environment

python3.12 -m venv .tpu_venv

Upgrade pip to handle modern packages

pip install --upgrade pip

Install JAX with TPU Support

pip install "jax[tpu]" -f <https://storage.googleapis.com/jax-releases/libtpu_releases.html>

Install the remaining libraries

pip install -r requirements_tpu.txt