Skip to content

Bypass mx.vmap for natively batched MLX linalg ops - #2399

Open
guillaume-osmo wants to merge 2 commits into
pymc-devs:mainfrom
guillaume-osmo:fix-mlx-batched-solve
Open

Bypass mx.vmap for natively batched MLX linalg ops#2399
guillaume-osmo wants to merge 2 commits into
pymc-devs:mainfrom
guillaume-osmo:fix-mlx-batched-solve

Commits

Commits on Aug 26, 2026