Index _ | A | B | C | D | E | F | G | I | K | L | M | N | P | R | S | T | U | V | W _ __call__() (ContrastiveLoss method) (ElboLoss method) (MaximumLikelihoodLoss method) A AbstractBijection (class in flowjax.bijections) AbstractDistribution (class in flowjax.distributions) AbstractLocScaleDistribution (class in flowjax.distributions) AbstractTransformed (class in flowjax.distributions) AbstractUnwrappable (class in flowjax.wrappers) AdditiveCondition (class in flowjax.bijections) Affine (class in flowjax.bijections) args (Parameterize attribute) B Beta (class in flowjax.distributions) block_neural_autoregressive_flow() (in module flowjax.flows) BlockAutoregressiveNetwork (class in flowjax.bijections) C Cauchy (class in flowjax.distributions) Chain (class in flowjax.bijections) Concatenate (class in flowjax.bijections) cond_ndim (AbstractDistribution property) ContrastiveLoss (class in flowjax.train.losses) Coupling (class in flowjax.bijections) coupling_flow() (in module flowjax.flows) covariance (MultivariateNormal property) D derivative() (RationalQuadraticSpline method) df (StudentT property) distribution_to_numpyro() (in module flowjax.experimental.numpyro) E ElboLoss (class in flowjax.train.losses) EmbedCondition (class in flowjax.bijections) Exp (class in flowjax.bijections) Exponential (class in flowjax.distributions) F fit_to_data() (in module flowjax.train) fit_to_key_based_loss() (in module flowjax.train) Flip (class in flowjax.bijections) flowjax.bijections module flowjax.distributions module flowjax.experimental.numpyro module flowjax.flows module flowjax.train.losses module flowjax.wrappers module fn (Parameterize attribute) G Gamma (class in flowjax.distributions) get_planar() (Planar method) Gumbel (class in flowjax.distributions) I Identity (class in flowjax.bijections) inv_scan_fn() (MaskedAutoregressive method) inverse() (AbstractBijection method) inverse_and_log_det() (AbstractBijection method) Invert (class in flowjax.bijections) K kwargs (Parameterize attribute) L Laplace (class in flowjax.distributions) LeakyTanh (class in flowjax.bijections) loc (AbstractLocScaleDistribution property) Loc (class in flowjax.bijections) loc (MultivariateNormal property) log_prob() (AbstractDistribution method) Logistic (class in flowjax.distributions) LogNormal (class in flowjax.distributions) M masked_autoregressive_flow() (in module flowjax.flows) MaskedAutoregressive (class in flowjax.bijections) MaximumLikelihoodLoss (class in flowjax.train.losses) maxval (Uniform property) merge_chains() (Chain method) merge_transforms() (AbstractTransformed method) minval (Uniform property) module flowjax.bijections flowjax.distributions flowjax.experimental.numpyro flowjax.flows flowjax.train.losses flowjax.wrappers MultivariateNormal (class in flowjax.distributions) N ndim (AbstractDistribution property) non_trainable() (in module flowjax.wrappers) NonTrainable (class in flowjax.wrappers) Normal (class in flowjax.distributions) num_samples (ElboLoss attribute) P Parameterize (class in flowjax.wrappers) Partial (class in flowjax.bijections) Permute (class in flowjax.bijections) Planar (class in flowjax.bijections) planar_flow() (in module flowjax.flows) Power (class in flowjax.bijections) R RationalQuadraticSpline (class in flowjax.bijections) register_params() (in module flowjax.experimental.numpyro) Reshape (class in flowjax.bijections) S sample() (AbstractDistribution method) (in module flowjax.experimental.numpyro) sample_and_log_prob() (AbstractDistribution method) scale (AbstractLocScaleDistribution property) Scale (class in flowjax.bijections) scale (WeightNormalization attribute) Scan (class in flowjax.bijections) Sigmoid (class in flowjax.bijections) SoftPlus (class in flowjax.bijections) Stack (class in flowjax.bijections) StandardNormal (class in flowjax.distributions) step() (in module flowjax.train) stick_the_landing (ElboLoss attribute) StudentT (class in flowjax.distributions) T Tanh (class in flowjax.bijections) target (ElboLoss attribute) transform() (AbstractBijection method) transform_and_log_det() (AbstractBijection method) Transformed (class in flowjax.distributions) tree (NonTrainable attribute) triangular_spline_flow() (in module flowjax.flows) TriangularAffine (class in flowjax.bijections) U Uniform (class in flowjax.distributions) unwrap() (AbstractUnwrappable method) (in module flowjax.wrappers) (NonTrainable method) (Parameterize method) (WeightNormalization method) V Vmap (class in flowjax.bijections) VmapMixture (class in flowjax.distributions) W weight (WeightNormalization attribute) WeightNormalization (class in flowjax.wrappers)