torch>=2.0.0
numpy
ipython
ipykernel