README.md
setup.cfg
setup.py
tpu_star/__init__.py
tpu_star/colab_utils.py
tpu_star/utils.py
tpu_star.egg-info/PKG-INFO
tpu_star.egg-info/SOURCES.txt
tpu_star.egg-info/dependency_links.txt
tpu_star.egg-info/requires.txt
tpu_star.egg-info/top_level.txt
tpu_star/datasets/__init__.py
tpu_star/datasets/mnist.py
tpu_star/experiment/__init__.py
tpu_star/experiment/base.py
tpu_star/experiment/metrics.py
tpu_star/experiment/torch_gpu.py
tpu_star/experiment/torch_tpu.py