First native Apple Silicon (MLX) port of Diffusion Policy (RSS 2023 Best Paper). 6 policy variants, 472 tests, Metal GPU verified. Train and run visuomotor diffusion policies on M-series — no CUDA required.
robotics imitation-learning robot-learning mlx apple-silicon diffusion-policy metal-gpu visuomotor-policy pytorch-to-mlx action-diffusion
-
Updated
Mar 15, 2026 - Python