deepmind/synjax: SynJax is a neural network library for JAX structured probability distributions
SynJax is a neural network library for JAX structured probability distributions
Category: math
Language: Python
Topics: distributions, jax, probability