Lightning-AI / pytorch-lightning UNCLAIMED

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

0 0 0 Python

[TPU] Add testing matrix with PJRT (#17368)

* Replace GKE in CI with manual gcloud usage

* Fix XRT test

* Reduce timeout to 35 minutes

* [TPU] Run tests with PJRT

* runtime as part of the job name

* CHANGELOG

* Update for app too

Carlos Mocholí committed 3y ago

05b481e3ae4ee687d87117d2c5b2a561b9591ffd

Parent: 856b29f

Committed by GitHub <noreply@github.com> on 4/14/2023, 2:39:13 PM