torch torchdiffeq torchsde omegaconf einops open-clip-torch transformers safetensors pytorch_lightning aiohttp accelerate python-socketio