/tpu_pod_launcher

Primary LanguagePython

Launch Experiments on TPU Pods with a Python interface

Installation

pip install git+https://github.com/Sea-Snell/tpu_pod_launcher.git@main

See the example for usage.

More documentation coming soon.