flowjax

github.comtool

Distributions and normalizing flows built as equinox modules.

from

JAX

added

2026-10-10

likes

0

JAX › Libraries > New Libraries: “Distributions and normalizing flows built as equinox modules.”