torch
safetensors
numpy
pytorch-lightning