PyTorch
Published in

PyTorch

How PyTorch Lightning became the first ML framework to run continuous integration on TPUs

Co-authored with Jirka Borovec, Lead contributor for PyTorch Lightning and Zach Cain, ML engineer @ Google

As PyTorch Lightning adoption continues to grow, we continuously evolve our testing suite to ensure that the companies and AI research labs that build their AI systems on PyTorch Lightning have a reliable and robust codebase. In preparation for our upcoming V1, we have taken a major step in our support for training on TPUs.

--

--

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store