Google open-sources TorchTPU, a native PyTorch back end for TPUs

Contribute to google-pytorch/torch_tpu development by creating an account on GitHub.

AI Summary

The GitHub project google-pytorch/torch_tpu provides a backend for running PyTorch models on Google's Tensor Processing Units (TPUs). It features custom ATen kernels, a compilation cache for optimized execution, and supports distributed training. The repository includes examples for various models and a user guide, with installation instructions compatible with Python 3.11 and above.

Read Original → · Discuss with AI → · Share →
← Back to news