numpy
torch
typing-extensions
