Have you ever wanted to train your fastai/pytorch models using TPUs?
I just released an early version (0.0.8) of the fastai_xla_extensions package.
Its primarily focused on vision models right now and you can
test it very easily - it comes with sample notebooks you can run on Colab!
The goal is that it should run existing fastai notebooks with very little modifications (except additional package installs of course), but for now its mostly the
fastai.vision packages I’ve tested.
Hope you try it out!
(and star the github project if you like it!)