flaxcrf (pypi): Conditional random field in JAX
Keywords: jax