numpy
torch
safetensors
pytorch_lightning