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

Eden Afek
Eden Afek
Jul 10, 2020 · 5 min read

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.