torch
safetensors
pytorch-lightning