Yes, that is confusing as the intro in the README says "SynJax is a neural network library for JAX structured probability distributions". However, contrary to

Yes, that is confusing as the intro in the README says "SynJax is a neural network library for JAX structured probability distributions". However, contrary to my expectations after reading that, they do not have lots primitives to mix neural and probabilistic models. They seem to be heading in that direction, with e.g. connectionist temporal classification (CTC). Still very interesting library, seems to provide massive efficiency gains for difficult generative models such as PCFGs.