google-pytorch / torch_tpu

Provides a PyTorch backend that runs PyTorch models and operations on Google's Tensor Processing Units (TPUs).

microsoft / onnxruntime

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator