NuX:使用JAX标准化流 源码

NuX-使用JAX标准化流 什么是NuX? NuX是用于使用构建的库。 什么是规范化流量? 归一化流使用数据学习未知概率密度函数上的参数模型。 我们假设数据点是从未知分布p(x)采样的。 归一化流使用最大似然学习来学习真实数据分布的参数近似。 可以从中高效地学习学习到的分布,并具有可以准确评估的对数似然。 为什么要使用NuX? 使用NuX可以轻松构建,训练和评估标准化流 import nux import jax import jax . numpy as jnp key = jax . random . PRNGKey ( 0 ) # Build a dummy dataset x_t