Thanks to the awesome work by the PyTorch and XLA team, we were able to get TPU support fully working out of the box with PyTorch Lightning.
Check out this nifty guide on going from PyTorch to PyTorch Lightning 30

  
      twitter.com
  
  
    

PyTorchLightning (PyTorchLightnin) 19

 You can now run @PyTorch code on TPUs and GPUs with Lightning... wait for it... 

WITHOUT CODE CHANGES!!! 

Retweet if this is useful to you! https://t.co/vyfFeAl5Yq



  11:42 PM - 26 Feb 2020
    
      
        
       620
    
    
      
        
       200