torch torchdiffeq torchsde einops open-clip-torch transformers safetensors pytorch_lightning aiohttp accelerate pyyaml