scipy
torch
typing-extensions