wai.annotations.core>=0.1.1
tensorflow<2.8.0,>=2.6.0
numpy
