feat(mlx): pt.random support with mlx backend - #1979

Open

williambdean wants to merge 15 commits into

pymc-devs:mainfrom

williambdean:mlx-random

Commits on Aug 27, 2026

add currently supported random module
williambdean
committed
648411c
add test suite
williambdean
committed
ac3b831
handle rng input
williambdean
committed
579941d
address review: shared rng dance in linker, xor-fold 128-bit pcg64 seed
williambdean
committed
9a58850
fix import after core.py renamed to tensor_basic.py
williambdean
committed
2688290
address review: bernoulli shape boilerplate, mvnormal decomposition methods, permutation error at dispatch time
williambdean
committed
0a93349
fix ruff isort: declare pytensor and tests as known-first-party
williambdean
committed
c2c4a31
fix import: pytensor.compile.function removed in v3 refactor
williambdean
committed
bd62f61
leverage shape
williambdean
committed
f069f62
Add LogNormal, HalfNormal, Exponential, Logistic, Cauchy MLX dispatches
williambdean
committed
b6e8fab
address review: consolidate loc-scale dispatches and fix RNG edge cases

authored and
williambdean
committed
d587333
fix mypy no-any-return and trim over-explanatory comments

authored and
williambdean
committed
eaf7d33
address review: consolidate loc-scale dispatch, parametrize shape tests
williambdean
committed
5ee96d6
fix: use float dtype family in test assertions for floatX compat
williambdean
committed
c783bd4

Commits on Aug 28, 2026

sample MLX randoms in float32 and cast to output dtype
williambdean
committed
d6f1884