/TPU-Star

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

Primary LanguagePythonApache License 2.0Apache-2.0

TPU-Star

Toolkit for quickly and easy prototyping on TPU and GPU. This repo was created by @shonenkov and is used for own research purposes without documentation, because used training mechanisms is widely known and well described in PyTorch. If you have found some link or pip library with this repo in my papers or tutorials - so sorry :D you can try to copy necessary parts of code or use another toolkit

PyTorch XLA Build Status Coverage Status Apache license

Installing

  1. Using pip:
pip install tpu-star==0.0.1-rc10

Examples

  • tpu_star & MNIST : Open In Colab