README.md
setup.cfg
setup.py
tpu_star/__init__.py
tpu_star/colab_utils.py
tpu_star/experiment.py
tpu_star/metrics.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