Index _ | A | B | C | D | E | F | G | I | L | M | N | P | R | S | T | U | V _ __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) AdditiveCondition (class in flowjax.bijections) Affine (class in flowjax.bijections) 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 G Gamma (class in flowjax.distributions) get_planar() (Planar method) Gumbel (class in flowjax.distributions) I Identity (class in flowjax.bijections) Indexed (class in flowjax.bijections) inv_scan_fn() (MaskedAutoregressive method) inverse() (AbstractBijection method) inverse_and_log_det() (AbstractBijection method) Invert (class in flowjax.bijections) 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 MultivariateNormal (class in flowjax.distributions) N ndim (AbstractDistribution property) Normal (class in flowjax.distributions) num_samples (ElboLoss attribute) NumericalInverse (class in flowjax.bijections) P 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) 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) triangular_spline_flow() (in module flowjax.flows) TriangularAffine (class in flowjax.bijections) U Uniform (class in flowjax.distributions) V Vmap (class in flowjax.bijections) VmapMixture (class in flowjax.distributions)
