JAX now runs on Cloud TPUs (with example collab notebooks)github.com/google2 pointsArtWomb7 years ago