a quick tutorial for getting set up with TPUs - View it on GitHub