Pulse
0
Velocity
Stars
0
Google's PyTorch extension for running on TPUs.
Enables PyTorch models to leverage Google's Tensor Processing Units for accelerated training and inference.
Run your PyTorch models faster on Google's powerful TPUs.
Why Trending
Accelerating ML workloads on Google's custom hardware.
Target Audience
ML researchers and engineers using PyTorch.
Similar Projects