matplotlib
numpy
tensorflow==2.5.3
tensorflow_datasets