build(torch): Update base/`torch`/`-vision`/`-audio`, DeepSpeed, & Apex

[Eta0]

Version Updates

Updates components to:

• Newer nccl-tests base images
• torch v2.1.1
• torchvision v0.16.1
• torchaudio v2.1.1
• DeepSpeed v0.12.4
• NVIDIA Apex 2386a91

[github-actions[bot]]

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/7133712240 Image: ghcr.io/coreweave/ml-containers/torch:es-update-extras-4af52b8-base-cuda12.1.1-torch2.1.1-vision0.16.1-audio2.1.1

[github-actions[bot]]

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/7133712240 Image: ghcr.io/coreweave/ml-containers/torch:es-update-extras-4af52b8-base-cuda11.8.0-torch2.1.1-vision0.16.1-audio2.1.1

[github-actions[bot]]

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/7133712244 Image: ghcr.io/coreweave/ml-containers/torch:es-update-extras-4af52b8-nccl-cuda12.0.1-nccl2.19.3-1-torch2.1.1-vision0.16.1-audio2.1.1

[github-actions[bot]]

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/7133712244 Image: ghcr.io/coreweave/ml-containers/torch:es-update-extras-4af52b8-nccl-cuda11.8.0-nccl2.16.5-1-torch2.1.1-vision0.16.1-audio2.1.1

[github-actions[bot]]

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/7133712240 Image: ghcr.io/coreweave/ml-containers/torch:es-update-extras-4af52b8-base-cuda12.0.1-torch2.1.1-vision0.16.1-audio2.1.1

[github-actions[bot]]

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/7133712240 Image: ghcr.io/coreweave/ml-containers/torch:es-update-extras-4af52b8-base-cuda12.2.2-torch2.1.1-vision0.16.1-audio2.1.1

[github-actions[bot]]

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/7133712244 Image: ghcr.io/coreweave/ml-containers/torch:es-update-extras-4af52b8-nccl-cuda12.1.1-nccl2.18.3-1-torch2.1.1-vision0.16.1-audio2.1.1

[github-actions[bot]]

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/7133712244 Image: ghcr.io/coreweave/ml-containers/torch:es-update-extras-4af52b8-nccl-cuda12.2.2-nccl2.19.3-1-torch2.1.1-vision0.16.1-audio2.1.1

[github-actions[bot]]

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/7133712256 Image: ghcr.io/coreweave/ml-containers/nightly-torch:es-update-extras-4af52b8-base-2023.12.07.20-cuda12.1.1-torch2.2.0a0-vision0.17.0a0-audio2.2.0a0

[github-actions[bot]]

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/7133712256 Image: ghcr.io/coreweave/ml-containers/nightly-torch:es-update-extras-4af52b8-base-2023.12.07.20-cuda12.0.1-torch2.2.0a0-vision0.17.0a0-audio2.2.0a0

[github-actions[bot]]

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/7133712256 Image: ghcr.io/coreweave/ml-containers/nightly-torch:es-update-extras-4af52b8-base-2023.12.07.20-cuda11.8.0-torch2.2.0a0-vision0.17.0a0-audio2.2.0a0

[github-actions[bot]]

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/7133712256 Image: ghcr.io/coreweave/ml-containers/nightly-torch:es-update-extras-4af52b8-base-2023.12.07.20-cuda12.2.2-torch2.2.0a0-vision0.17.0a0-audio2.2.0a0

[github-actions[bot]]

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/7137942835 Image: ghcr.io/coreweave/ml-containers/torch-extras:es-update-extras-16ca467-nccl-cuda12.0.1-nccl2.18.5-1-torch2.0.1-vision0.15.2-audio2.0.2-flash_attn2.3.6

[github-actions[bot]]

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/7137942835 Image: ghcr.io/coreweave/ml-containers/torch-extras:es-update-extras-16ca467-nccl-cuda11.8.0-nccl2.16.2-1-torch2.0.1-vision0.15.2-audio2.0.2-flash_attn2.3.6

[github-actions[bot]]

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/7137942835 Image: ghcr.io/coreweave/ml-containers/torch-extras:es-update-extras-16ca467-base-cuda12.1.1-torch2.0.1-vision0.15.2-audio2.0.2-flash_attn2.3.6

[github-actions[bot]]

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/7137942835 Image: ghcr.io/coreweave/ml-containers/torch-extras:es-update-extras-16ca467-nccl-cuda12.2.2-nccl2.18.5-1-torch2.0.1-vision0.15.2-audio2.0.2-flash_attn2.3.6

[github-actions[bot]]

@Eta0 Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/7137942835 Image: ghcr.io/coreweave/ml-containers/torch-extras:es-update-extras-16ca467-nccl-cuda12.1.1-nccl2.18.3-1-torch2.0.1-vision0.15.2-audio2.0.2-flash_attn2.3.6