huangziwei
commented
1 year ago jax.scipy.stats.vonmises currently (jax + jaxlib == 0.4.2) doesn't take loc and scale as arguments. Is it intended?
Open carlosgmartin opened 2 years ago
huangziwei
commented
1 year ago jax.scipy.stats.vonmises currently (jax + jaxlib == 0.4.2) doesn't take loc and scale as arguments. Is it intended?
Add the following directional distributions to jax.random (for sampling) and/or jax.scipy.stats (for probability density function evaluation):
vonmises: von Mises distributionwrapped_normal: wrapped normal distributionwrapped_cauchy: wrapped Cauchy distributionvonmises_fisher: von Mises–Fisher distributionpower_spherical: power spherical distribution (github, tfp, rstudio)