Ensure compatibility with TPUs
Closed this issue ยท 1 comments
SeanNaren commented
๐ Feature
I haven't tried to see if TPUs work out of the box. We should ensure that the code runs as expected using the tasks.
stale commented
This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.