SynJax: Jax library for efficient probabilistic modeling of structured objectsgithub.com/deepmind1 pointsapsan3 years ago