torch torchvision torchaudio pytorch-lightning einops omegaconf deepspeed torchmetrics webdataset accelerate tensorboard PyMCubes trimesh rembg transformers diffusers bitsandbytes imageio[ffmpeg] xatlas plyfile xformers git+https://github.com/NVlabs/nvdiffrast/ huggingface-hub httpx flask pillow safetensors