6/ SynJax – a library that provides an efficient vectorized implementation of inference algorithms for structured distributions; enables building large-scale differentiable models that model structure in data like segmentation and constituency trees.

→ View original post on X — @dair_ai,