[MLX] Upgrade to Torch 2.13/MLX 0.32+ and redesign the Torch-MLX tensor bridge (#32984)

Co-authored-by: Alex Nails <alex.nails@radixark.ai>
2026-08-21 18:51:42 -07:00
parent 3b5909de0e
commit d90318b3e2
36 changed files with 1695 additions and 343 deletions
@@ -167,14 +167,15 @@ diffusion_musa = [
  "vsa==0.0.4",
 ]

-# https://docs.sglang.io/platforms/mps.md
+# https://docs.sglang.io/hardware-platforms/apple_metal
 srt_mps = [
-    "mlx",
+    "mlx>=0.32.0",
    "mlx-lm",
    "sglang[runtime_common]",
-    "torch==2.11.0",
+    "torch==2.13.0",
    "torchaudio==2.11.0",
-    "torchvision",
+    "torchcodec==0.15.0",
+    "torchvision==0.28.0",
 ]

 diffusion_mps = [