Pyro distribution in random flax

Hi everyone.

How can I use a pyro distribution as a prior in random_flax_module? Since the prior needs to be a numpyro distribution.

I will be grateful for your help.

Regards

you cannot. pyro and numpyro are distinct packages and are not interoperable.

1 Like