torch_tpu

What does it do?

It provides a PyTorch backend for Google's Tensor Processing Units (TPUs), which lets PyTorch models and operations run on TPU devices. [1] The project includes custom ATen kernels, which are operation implementations for PyTorch, and a compilation cache for optimizing execution. [2] It includes examples for distributed training and model inference. [3]

Activity, over time

Daily observations from the public event archive.

Attention

Daily stars & forks

Loading attention chart

Star accumulation

Cumulative stars

Loading star accumulation chart

Pull requests

Daily opens & merges

Loading pull requests chart

Conversations

Daily issues & comments

Loading conversations chart

Community

Daily accounts & reviews

Loading community chart

Releases

Daily releases

Loading releases chart

Latest stargazers

The last 10 distinct accounts we observed this year.

Most active participants

PR, issue, comment & review events in the past year.

Packages

1
PackageVersionDownloads · 30dTrend · 30d
torch_tpuv0.2.0
v0.2.0
≥7.6K

README badges

Embed a badge in your README or website.

Live mentions · last 30 days
google-pytorch/torch_tpu mentions in the last 30 days