Aug. 29, 2023, 2:24 a.m. | /u/impromptued

Machine Learning www.reddit.com

See title. My impression has always been that PyTorch for TPU is an in-name only functionality, but I'm curious about first-hand experience from those who have used it after PyTorch 2.0+.

Bonus question: has anyone used PyTorch Lightning for running on TPU? If so, how was the experience?

bonus experience machinelearning pytorch pytorch 2.0 pytorch lightning running tpu

AI Engineer Intern, Agents

@ Occam AI | US

AI Research Scientist

@ Vara | Berlin, Germany and Remote

Data Architect

@ University of Texas at Austin | Austin, TX

Data ETL Engineer

@ University of Texas at Austin | Austin, TX

Lead GNSS Data Scientist

@ Lurra Systems | Melbourne

Lead Data Modeler

@ Sherwin-Williams | Cleveland, OH, United States