Primary symbols
oryx
oryx.bijectors
oryx.bijectors.AbsoluteValue
oryx.bijectors.Affine
oryx.bijectors.AffineLinearOperator
oryx.bijectors.AffineScalar
oryx.bijectors.Ascending
oryx.bijectors.AutoregressiveNetwork
oryx.bijectors.Bijector
oryx.bijectors.Blockwise
oryx.bijectors.Chain
oryx.bijectors.CholeskyOuterProduct
oryx.bijectors.CholeskyToInvCholesky
oryx.bijectors.Composition
oryx.bijectors.CorrelationCholesky
oryx.bijectors.Cumsum
oryx.bijectors.Exp
oryx.bijectors.Expm1
oryx.bijectors.FillScaleTriL
oryx.bijectors.FillTriangular
oryx.bijectors.FrechetCDF
oryx.bijectors.GeneralizedExtremeValueCDF
oryx.bijectors.GeneralizedPareto
oryx.bijectors.GompertzCDF
oryx.bijectors.GumbelCDF
oryx.bijectors.Identity
oryx.bijectors.Inline
oryx.bijectors.Invert
oryx.bijectors.IteratedSigmoidCentered
oryx.bijectors.JointMap
oryx.bijectors.KumaraswamyCDF
oryx.bijectors.LambertWTail
oryx.bijectors.Log
oryx.bijectors.Log1p
oryx.bijectors.MaskedAutoregressiveFlow
oryx.bijectors.MatrixInverseTriL
oryx.bijectors.MatvecLU
oryx.bijectors.MoyalCDF
oryx.bijectors.NormalCDF
oryx.bijectors.Ordered
oryx.bijectors.Pad
oryx.bijectors.Permute
oryx.bijectors.Power
oryx.bijectors.PowerTransform
oryx.bijectors.RationalQuadraticSpline
oryx.bijectors.RayleighCDF
oryx.bijectors.RealNVP
oryx.bijectors.Reciprocal
oryx.bijectors.Reshape
oryx.bijectors.Restructure
oryx.bijectors.Scale
oryx.bijectors.ScaleMatvecDiag
oryx.bijectors.ScaleMatvecLU
oryx.bijectors.ScaleMatvecLinearOperator
oryx.bijectors.ScaleMatvecLinearOperatorBlock
oryx.bijectors.ScaleMatvecTriL
oryx.bijectors.Shift
oryx.bijectors.ShiftedGompertzCDF
oryx.bijectors.Sigmoid
oryx.bijectors.Sinh
oryx.bijectors.SinhArcsinh
oryx.bijectors.SoftClip
oryx.bijectors.Softfloor
oryx.bijectors.SoftmaxCentered
oryx.bijectors.Softplus
oryx.bijectors.Softsign
oryx.bijectors.Split
oryx.bijectors.Square
oryx.bijectors.Tanh
oryx.bijectors.TransformDiagonal
oryx.bijectors.Transpose
oryx.bijectors.WeibullCDF
oryx.bijectors.masked_autoregressive_default_template
oryx.bijectors.masked_dense
oryx.bijectors.real_nvp_default_template
oryx.core
oryx.core.FlatPrimitive
oryx.core.HigherOrderPrimitive
oryx.core.NonInvertibleError
oryx.core.Pytree
oryx.core.call_bind
oryx.core.custom_inverse
oryx.core.harvest
oryx.core.ildj
oryx.core.interpreters
oryx.core.interpreters.harvest
oryx.core.interpreters.harvest.HarvestTrace
oryx.core.interpreters.harvest.HarvestTracer
oryx.core.interpreters.harvest.harvest
oryx.core.interpreters.harvest.nest
oryx.core.interpreters.harvest.plant
oryx.core.interpreters.harvest.reap
oryx.core.interpreters.harvest.sow
oryx.core.interpreters.inverse
oryx.core.interpreters.inverse.core
oryx.core.interpreters.inverse.core.InverseAndILDJ
oryx.core.interpreters.inverse.core.inverse
oryx.core.interpreters.inverse.core.inverse_and_ildj
oryx.core.interpreters.inverse.core.register_binary
oryx.core.interpreters.inverse.core.register_elementwise
oryx.core.interpreters.inverse.custom_inverse
oryx.core.interpreters.inverse.custom_inverse.NonInvertibleError
oryx.core.interpreters.inverse.custom_inverse.custom_inverse
oryx.core.interpreters.inverse.rules
oryx.core.interpreters.inverse.slice
oryx.core.interpreters.log_prob
oryx.core.interpreters.log_prob.LogProbRules
oryx.core.interpreters.log_prob.log_prob
oryx.core.interpreters.propagate
oryx.core.interpreters.propagate.Cell
oryx.core.interpreters.propagate.Environment
oryx.core.interpreters.propagate.Equation
oryx.core.interpreters.propagate.propagate
oryx.core.interpreters.unzip
oryx.core.interpreters.unzip.UnzipTrace
oryx.core.interpreters.unzip.UnzipTracer
oryx.core.interpreters.unzip.VariableError
oryx.core.interpreters.unzip.unzip
oryx.core.inverse
oryx.core.inverse_and_ildj
oryx.core.log_prob
oryx.core.nest
oryx.core.plant
oryx.core.ppl
oryx.core.ppl.LogProbFunction
oryx.core.ppl.Program
oryx.core.ppl.block
oryx.core.ppl.conditional
oryx.core.ppl.graph_replace
oryx.core.ppl.intervene
oryx.core.ppl.joint_log_prob
oryx.core.ppl.joint_sample
oryx.core.ppl.log_prob
oryx.core.ppl.nest
oryx.core.ppl.random_variable
oryx.core.ppl.rv
oryx.core.ppl.transformations
oryx.core.ppl.transformations.block
oryx.core.ppl.transformations.conditional
oryx.core.ppl.transformations.graph_replace
oryx.core.ppl.transformations.intervene
oryx.core.ppl.transformations.joint_log_prob
oryx.core.ppl.transformations.joint_sample
oryx.core.ppl.transformations.log_prob
oryx.core.ppl.transformations.nest
oryx.core.ppl.transformations.random_variable
oryx.core.ppl.transformations.rv
oryx.core.primitive
oryx.core.primitive.FlatPrimitive
oryx.core.primitive.HigherOrderPrimitive
oryx.core.primitive.call_bind
oryx.core.primitive.tie_all
oryx.core.primitive.tie_in
oryx.core.pytree
oryx.core.pytree.Pytree
oryx.core.reap
oryx.core.serialize
oryx.core.serialize.deserialize
oryx.core.serialize.from_tuple
oryx.core.serialize.serialize
oryx.core.serialize.to_tuple
oryx.core.sow
oryx.core.state
oryx.core.state.ArraySpec
oryx.core.state.Module
oryx.core.state.Shape
oryx.core.state.api
oryx.core.state.api.ArraySpec
oryx.core.state.api.call
oryx.core.state.api.call_and_update
oryx.core.state.api.init
oryx.core.state.api.make_array_spec
oryx.core.state.api.spec
oryx.core.state.api.update
oryx.core.state.assign
oryx.core.state.call
oryx.core.state.call_and_update
oryx.core.state.function
oryx.core.state.function.FunctionModule
oryx.core.state.init
oryx.core.state.make_array_spec
oryx.core.state.module
oryx.core.state.module.Module
oryx.core.state.module.assign
oryx.core.state.module.variable
oryx.core.state.spec
oryx.core.state.update
oryx.core.state.variable
oryx.core.tie_all
oryx.core.tie_in
oryx.core.trace_util
oryx.core.trace_util.get_dynamic_context
oryx.core.trace_util.get_shaped_aval
oryx.core.trace_util.new_dynamic_context
oryx.core.trace_util.pv_like
oryx.core.trace_util.stage
oryx.core.trace_util.trees
oryx.core.unzip
oryx.distributions
oryx.distributions.Autoregressive
oryx.distributions.BatchReshape
oryx.distributions.Bates
oryx.distributions.Bernoulli
oryx.distributions.Beta
oryx.distributions.BetaBinomial
oryx.distributions.Binomial
oryx.distributions.Blockwise
oryx.distributions.Categorical
oryx.distributions.Cauchy
oryx.distributions.Chi
oryx.distributions.Chi2
oryx.distributions.CholeskyLKJ
oryx.distributions.Deterministic
oryx.distributions.Dirichlet
oryx.distributions.DirichletMultinomial
oryx.distributions.Distribution
oryx.distributions.DoublesidedMaxwell
oryx.distributions.Empirical
oryx.distributions.ExpGamma
oryx.distributions.ExpInverseGamma
oryx.distributions.ExpRelaxedOneHotCategorical
oryx.distributions.Exponential
oryx.distributions.ExponentiallyModifiedGaussian
oryx.distributions.Gamma
oryx.distributions.GammaGamma
oryx.distributions.GaussianProcess
oryx.distributions.GaussianProcessRegressionModel
oryx.distributions.GeneralizedExtremeValue
oryx.distributions.GeneralizedNormal
oryx.distributions.GeneralizedPareto
oryx.distributions.Geometric
oryx.distributions.Gumbel
oryx.distributions.HalfCauchy
oryx.distributions.HalfNormal
oryx.distributions.HalfStudentT
oryx.distributions.HiddenMarkovModel
oryx.distributions.Horseshoe
oryx.distributions.Independent
oryx.distributions.InverseGamma
oryx.distributions.InverseGaussian
oryx.distributions.JohnsonSU
oryx.distributions.JointDistribution
oryx.distributions.JointDistributionCoroutine
oryx.distributions.JointDistributionCoroutine.Root
oryx.distributions.JointDistributionCoroutineAutoBatched
oryx.distributions.JointDistributionCoroutineAutoBatched.Root
oryx.distributions.JointDistributionNamed
oryx.distributions.JointDistributionNamedAutoBatched
oryx.distributions.JointDistributionSequential
oryx.distributions.JointDistributionSequentialAutoBatched
oryx.distributions.Kumaraswamy
oryx.distributions.LKJ
oryx.distributions.Laplace
oryx.distributions.LinearGaussianStateSpaceModel
oryx.distributions.LogLogistic
oryx.distributions.LogNormal
oryx.distributions.Logistic
oryx.distributions.LogitNormal
oryx.distributions.MixtureSameFamily
oryx.distributions.Moyal
oryx.distributions.Multinomial
oryx.distributions.MultivariateNormalDiag
oryx.distributions.MultivariateNormalDiagPlusLowRank
oryx.distributions.MultivariateNormalFullCovariance
oryx.distributions.MultivariateNormalLinearOperator
oryx.distributions.MultivariateNormalTriL
oryx.distributions.MultivariateStudentTLinearOperator
oryx.distributions.NegativeBinomial
oryx.distributions.Normal
oryx.distributions.OneHotCategorical
oryx.distributions.OrderedLogistic
oryx.distributions.PERT
oryx.distributions.Pareto
oryx.distributions.PlackettLuce
oryx.distributions.Poisson
oryx.distributions.PoissonLogNormalQuadratureCompound
oryx.distributions.PowerSpherical
oryx.distributions.ProbitBernoulli
oryx.distributions.QuantizedDistribution
oryx.distributions.RegisterKL
oryx.distributions.RelaxedBernoulli
oryx.distributions.RelaxedOneHotCategorical
oryx.distributions.ReparameterizationType
oryx.distributions.Sample
oryx.distributions.SinhArcsinh
oryx.distributions.Skellam
oryx.distributions.SphericalUniform
oryx.distributions.StoppingRatioLogistic
oryx.distributions.StudentT
oryx.distributions.StudentTProcess
oryx.distributions.TransformedDistribution
oryx.distributions.Triangular
oryx.distributions.TruncatedCauchy
oryx.distributions.TruncatedNormal
oryx.distributions.Uniform
oryx.distributions.VariationalGaussianProcess
oryx.distributions.VectorDeterministic
oryx.distributions.VectorExponentialDiag
oryx.distributions.VonMises
oryx.distributions.VonMisesFisher
oryx.distributions.Weibull
oryx.distributions.WishartLinearOperator
oryx.distributions.WishartTriL
oryx.distributions.independent_joint_distribution_from_structure
oryx.distributions.kl_divergence
oryx.distributions.mvn_conjugate_linear_update
oryx.distributions.normal_conjugates_known_scale_posterior
oryx.distributions.normal_conjugates_known_scale_predictive
oryx.distributions.quadrature_scheme_lognormal_gauss_hermite
oryx.distributions.quadrature_scheme_lognormal_quantiles
oryx.experimental
oryx.experimental.mcmc
oryx.experimental.mcmc.constrain
oryx.experimental.mcmc.hmc
oryx.experimental.mcmc.kernels
oryx.experimental.mcmc.kernels.hmc
oryx.experimental.mcmc.kernels.mala
oryx.experimental.mcmc.kernels.metropolis
oryx.experimental.mcmc.kernels.metropolis_hastings
oryx.experimental.mcmc.kernels.random_walk
oryx.experimental.mcmc.kernels.sample_chain
oryx.experimental.mcmc.mala
oryx.experimental.mcmc.metropolis
oryx.experimental.mcmc.metropolis_hastings
oryx.experimental.mcmc.random_walk
oryx.experimental.mcmc.sample_chain
oryx.experimental.mcmc.utils
oryx.experimental.mcmc.utils.constrain
oryx.experimental.nn
oryx.experimental.nn.AvgPooling
oryx.experimental.nn.BatchNorm
oryx.experimental.nn.Conv
oryx.experimental.nn.Deconv
oryx.experimental.nn.Dense
oryx.experimental.nn.Dropout
oryx.experimental.nn.Flatten
oryx.experimental.nn.Layer
oryx.experimental.nn.LayerParams
oryx.experimental.nn.LogSoftmax
oryx.experimental.nn.MaxPooling
oryx.experimental.nn.Relu
oryx.experimental.nn.Reshape
oryx.experimental.nn.Serial
oryx.experimental.nn.Softmax
oryx.experimental.nn.Softplus
oryx.experimental.nn.SumPooling
oryx.experimental.nn.Tanh
oryx.experimental.nn.Template
oryx.experimental.nn.base
oryx.experimental.nn.base.Layer
oryx.experimental.nn.base.LayerParams
oryx.experimental.nn.base.Template
oryx.experimental.nn.base.create_parameter
oryx.experimental.nn.combinator
oryx.experimental.nn.combinator.Serial
oryx.experimental.nn.convolution
oryx.experimental.nn.convolution.Conv
oryx.experimental.nn.convolution.Deconv
oryx.experimental.nn.convolution.conv_info
oryx.experimental.nn.core
oryx.experimental.nn.core.Activation
oryx.experimental.nn.core.Dense
oryx.experimental.nn.core.Dropout
oryx.experimental.nn.core.LogSoftmax
oryx.experimental.nn.core.Relu
oryx.experimental.nn.core.Softmax
oryx.experimental.nn.core.Softplus
oryx.experimental.nn.core.Tanh
oryx.experimental.nn.normalization
oryx.experimental.nn.normalization.BatchNorm
oryx.experimental.nn.pooling
oryx.experimental.nn.pooling.AvgPooling
oryx.experimental.nn.pooling.MaxPooling
oryx.experimental.nn.pooling.Pooling
oryx.experimental.nn.pooling.SumPooling
oryx.experimental.nn.reshape
oryx.experimental.nn.reshape.Flatten
oryx.experimental.nn.reshape.Reshape
oryx.experimental.optimizers
oryx.experimental.optimizers.adam
oryx.experimental.optimizers.gradient_descent
oryx.experimental.optimizers.noisy_sgd
oryx.experimental.optimizers.optimize
oryx.experimental.optimizers.optix
oryx.experimental.optimizers.optix.adam
oryx.experimental.optimizers.optix.add_noise
oryx.experimental.optimizers.optix.apply_every
oryx.experimental.optimizers.optix.chain
oryx.experimental.optimizers.optix.clip
oryx.experimental.optimizers.optix.clip_by_global_norm
oryx.experimental.optimizers.optix.global_norm
oryx.experimental.optimizers.optix.gradient_descent
oryx.experimental.optimizers.optix.noisy_sgd
oryx.experimental.optimizers.optix.optimize
oryx.experimental.optimizers.optix.rmsprop
oryx.experimental.optimizers.optix.scale
oryx.experimental.optimizers.optix.scale_by_adam
oryx.experimental.optimizers.optix.scale_by_rms
oryx.experimental.optimizers.optix.scale_by_schedule
oryx.experimental.optimizers.optix.scale_by_stddev
oryx.experimental.optimizers.optix.sgd
oryx.experimental.optimizers.optix.trace
oryx.experimental.optimizers.rmsprop
oryx.experimental.optimizers.sgd
oryx.internal
oryx.internal.test_util
oryx.internal.test_util.TestCase
oryx.internal.test_util.TestCase.failureException
oryx.util
oryx.util.get_summaries
oryx.util.summary
oryx.version