TorchTPU registers Google TPUs as standard hardware devices using PyTorch's PrivateUse1 interface, enabling eager execution without source code modifications or wrappers.

Activity

Activity (0)
Loading comments...