numpy<2
tensorflow==2.16.1
tf-keras==2.16.0
torch==2.11.0+cu126
torchvision==0.26.0+cu126