matplotlib
numpy
tensorflow==2.5.2
tensorflow_datasets