机器之心 (微信公众号)T2·88 pts
- TorchTPU elevates TPU to a native supported device in PyTorch, alongside GPU and CPU.
- Developers can deploy models via the device_map="tpu" parameter in Hugging Face Transformers v5.
- The initiative aims to eliminate engineering costs associated with the PyTorch/XLA translation layer and challenge CUDA's dominance.