numpy
pandas
scikit-learn
tensorflow>=2.14.0
tf-keras
tensorflow-probability~=0.22.1
wandb
