numpy~=1.19.2
tensorflow>=2.4.0
ray[tune]
matplotlib
wandb
focal_loss
pillow
