Hacker NewsHN submitter: ilreb
Original published:

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

Google has open-sourced TorchTPU, a native PyTorch back end designed for TPUs, allowing developers to run PyTorch models natively on TPU hardware.

Key points

  • •It is developed as a native back end to execute PyTorch models on TPUs.
  • •Development contributions are facilitated through the google-pytorch/torch_tpu repository on GitHub.
#Semiconductors & Infrastructure#LLM / Model Trends#Major Players#TorchTPU#Google
See more