← Latest products

TorchTPU

Running PyTorch Natively on TPUs at Google Scale. TorchTPU is Google's PyTorch-native backend for TPUs.

Visit cloud.google.comexternal site

Reactions (1)

Similar products

by description similarity

  1. Run PyTorch locally with a remote GPU backend

    github.com

    I integrated a remote GPU execution backend into PyTorch through the same system that custom hardware accelerators get integrated into PyTorch.

    Hacker News5 pts

  2. GoTorch

    news.ycombinator.com

    A Go Implementation of PyTorch

    Hacker News9 pts

  3. Graphlearn-for-PyTorch

    github.com

    distributed graph learning on PyTorch

    Hacker News9 pts

  4. Rgpu

    github.com

    a PyTorch device whose tensors live on a remote GPU

    Hacker News57 pts

  5. Open-Sourcing Google's Lattice Models in PyTorch

    github.com

    A PyTorch implementation of constrained optimization and modeling techniques

    Hacker News13 pts