search

aesara-devs / aehmc

An HMC/NUTS implementation in Aesara
MIT License
33 stars 6 forks source link

Track Numba compilation #83

Open rlouf opened 2 years ago

rlouf commented 2 years ago

I open this PR to track progress on using the Numba backend to compile the NUTS (and HMC) sampler. The test currently fails in make_numba_random_fn with 

TypeError: Numba does not support NumPy `Generator`s

Indeed, RandomStream is currently not supported by the Numba backend https://github.com/aesara-devs/aesara/issues/801

Closes #68.