pandas
scikit-learn
tensorflow>=2.15.1
keras
keras_tuner
torch
torch-geometric
optuna
tqdm

[datagen]
hls4ml
psutil
py-cpuinfo
pyparsing
tensorflow==2.15.1
tf-keras==2.15.1

[dev]
pre-commit
pytest
