SynJax: Structured Probability Distributions for JAX

Open in new window