numpy==1.21.5 tensorflow_model_optimization tf-models-official==2.7 psutil tensorboard==2.7 keras==2.7