Now support (and enforce) new style jax keys, i.e. update any jr.PRNGKey(seed) calls with jr.key(seed).
Improved documentation.
LogNormal now implemented as a transformed Normal distribution, rather than a transformed StandardNormal (neater structure, and nicer to do log_normal.base_dist.loc, to see the location parameter, rather than going into the Chain bijection previously used.
Changes:
jr.PRNGKey(seed)calls withjr.key(seed).LogNormalnow implemented as a transformed Normal distribution, rather than a transformedStandardNormal(neater structure, and nicer to dolog_normal.base_dist.loc, to see the location parameter, rather than going into theChainbijection previously used.