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

Bypass mx.vmap for natively batched MLX linalg ops (#2385)

a831932
Select commit
Loading
Failed to load commit list.