orenleung/jax_tpu

Docker image for running JAX on TPU VM, making it easy to take advantage of the power of TPUs for ML

Latest release: over 2 years ago - 43 downloads - 16 dependencies

More details on packages.ecosyste.ms - JSON

Versions