memory_profiler
numba
numpy
scipy
torch==2.*
