Lorenzo Rimella

Dirichlet distribution sampling issue when jit_compile=True

1

It seems that some seeds produce nans when sampling from a Dirichlet distribution. Any idea why? Example script below that was tested on Google Colab. ``` import numpy as np...