synjax
SynJax is a neural network library designed for JAX, emphasizing structured probability distributions such as Linear Chain CRF and Semi-Markov CRF. It supports essential operations like log-probabilities and entropy calculations, utilizing JAX transformations for optimized performance. SynJax provides easy installation aligned with JAX's guidelines and includes practical examples in its notebooks for effective learning.
