Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
语言:Python · GitHub 星标:48 · Fork:10
话题:jax · machine-learning · statistics
在 GitHub 上查看原仓库