XLA
Enabling PyTorch on XLA Devices (e.g. Google TPU)
About
PyTorch/XLA is a Python package that uses the XLA deep learning compiler to connect the PyTorch deep learning framework and Cloud TPUs. You can try it right now, for free, on a single Cloud TPU VM with Kaggle!
Open Source Health
- Stars
- 2,804
- Forks
- 571
- License
- Not stated
- Last commit
- 5 months ago
Resources & Links
Related Categories
Vendor
PyTorch
Publisher of Captum and Bayesian Optimization Library (botorch)
Quick Links
Open Source
More by PyTorch
Related Products
Vision
Datasets, Transforms and Models specific to Computer Vision
More from this vendor
TensorRT
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
More from this vendor
PyTorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
More from this vendor
Rclip
Semantic photo search for the command line
Top category match
ChineseNER
A neural network model for Chinese named entity recognition
Top category match