search

libffcv / ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)
https://ffcv.io
Apache License 2.0
2.79k stars 180 forks source link

Will FFCV work with PyTorch XLA with TPU? #303

Open opooladz opened 1 year ago

opooladz commented 1 year ago

Hi quick question,

Should we expect FFCV to work plug and play with TPUs if we are using PyTorch with XLA?

Thank you, Omead