numpy
matplotlib
scikit-learn
--extra-index-url https://download.pytorch.org/whl/nightly/cu126
--pre
torch
botorch
inferactively-pymdp