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