Resolve MLX integer arguments to Python ints for MLX 0.32.2 - #2424

Merged

jessegrabowski merged 8 commits into

pymc-devs:mainfrom

jessegrabowski:mlx-static-int-args

Sep 18, 2026

Commits on Sep 17, 2026

MLX: generalize the dynamic value error and expose coerce_to_int
jessegrabowski
committed
cb27cfc

Commits on Sep 18, 2026

MLX: resolve Reshape shape to Python ints

jessegrabowski
and
guillaume-osmo
committed
705461c
MLX: pass sort axis as a Python int

jessegrabowski
and
guillaume-osmo
committed
4861d5f
MLX: return int64 from argsort
jessegrabowski
committed
e5349db
MLX: resolve pad_width to Python int pairs

jessegrabowski
and
guillaume-osmo
committed
19c39c1
Compare eigh eigenvectors up to column sign in the MLX test
jessegrabowski
committed
0e3f86d
Lift the MLX upper pin in CI
jessegrabowski
committed
10595cb
MLX: cast argsort to the declared output dtype
jessegrabowski
committed
ceb9509