Add MLX backend support for Nutpie compilation - #254

Open

cetagostini wants to merge 16 commits into

pymc-devs:mainfrom

cetagostini:cetagostini/adding_mlx_backend

Commits on Oct 27, 2025

Add MLX backend support for Nutpie compilation
cetagostini
committed
0e95705
Update MLX dependency and JIT compile logp function
cetagostini
committed
d57f930
issue with tests
cetagostini
committed
f73cb55
Update test_pymc.py
cetagostini
committed
13b9a43

Commits on Oct 28, 2025

Changes
cetagostini
committed
f97d002
precommit
cetagostini
committed
7543d76

Commits on Oct 29, 2025

Create test_deterministic_sampling_mlx.txt
cetagostini
committed
905f61f
Update test_pymc.py
cetagostini
committed
ad3e2d6

Commits on Apr 28, 2026

Merge upstream/main into cetagostini/adding_mlx_backend
cetagostini
committed
0777c0d
fix(mlx): stabilise MLX backend and pin mlx<0.31
cetagostini
committed
fdeddfd
chore(mlx): trim narrative comments
cetagostini
committed
9ba876c

Commits on Jun 29, 2026

Commits on Jun 30, 2026

ci: keep local wheel preferred when installing dev/docs extras

cetagostini
and
claude
committed
4035c41
ci: stop pinning pytensor@main in dev constraints

cetagostini
and
claude
committed
584f115