dm-haiku>=0.0.3
tensorflow-datasets>=4.2.0
tensorflow>=2.4.0
