API

zuko.distributions	Parameterizable probability distributions.
zuko.flows	Parameterized flows and autoregressive transformations.
zuko.nn	Neural networks, layers and modules.
zuko.transforms	Parameterizable transformations.
zuko.utils	General purpose helpers.