dataclasses
fire
imageio
mnist-pipeline==0.2.1

[cpu]
tensorflow

[gpu]
tensorflow-gpu
