numpy
safetensors
torch
