/TPU-Star

PyTorch XLA Toolkit for quickly and easy prototyping on TPU (8 cores)

Primary LanguagePythonApache License 2.0Apache-2.0

Stargazers