Primary symbols
oryx
oryx.bijectors
oryx.bijectors.AbsoluteValue
oryx.bijectors.Ascending
oryx.bijectors.AutoCompositeTensorBijector
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.Householder
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.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.pack_sequence_as
oryx.bijectors.real_nvp_default_template
oryx.bijectors.tree_flatten
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.call_and_reap
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.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.effect_handler
oryx.core.ppl.effect_handler.make_effect_handler
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.make_effect_handler
oryx.core.ppl.nest
oryx.core.ppl.plate
oryx.core.ppl.plate_util
oryx.core.ppl.plate_util.make_plate
oryx.core.ppl.random_variable
oryx.core.ppl.rv
oryx.core.ppl.trace
oryx.core.ppl.trace_log_prob
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.distributions
oryx.distributions.Autoregressive
oryx.distributions.BatchBroadcast
oryx.distributions.BatchReshape
oryx.distributions.Bates
oryx.distributions.Bernoulli
oryx.distributions.Beta
oryx.distributions.BetaBinomial
oryx.distributions.BetaQuotient
oryx.distributions.Binomial
oryx.distributions.Blockwise
oryx.distributions.Categorical
oryx.distributions.Cauchy
oryx.distributions.Chi
oryx.distributions.Chi2
oryx.distributions.CholeskyLKJ
oryx.distributions.DeterminantalPointProcess
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.FiniteDiscrete
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.Inflated
oryx.distributions.InverseGamma
oryx.distributions.InverseGaussian
oryx.distributions.JohnsonSU
oryx.distributions.JointDistribution
oryx.distributions.JointDistribution.Root
oryx.distributions.JointDistributionCoroutine
oryx.distributions.JointDistributionCoroutine.Root
oryx.distributions.JointDistributionCoroutineAutoBatched
oryx.distributions.JointDistributionCoroutineAutoBatched.Root
oryx.distributions.JointDistributionNamed
oryx.distributions.JointDistributionNamed.Root
oryx.distributions.JointDistributionNamedAutoBatched
oryx.distributions.JointDistributionNamedAutoBatched.Root
oryx.distributions.JointDistributionSequential
oryx.distributions.JointDistributionSequential.Root
oryx.distributions.JointDistributionSequentialAutoBatched
oryx.distributions.JointDistributionSequentialAutoBatched.Root
oryx.distributions.Kumaraswamy
oryx.distributions.LKJ
oryx.distributions.LambertWDistribution
oryx.distributions.LambertWNormal
oryx.distributions.Laplace
oryx.distributions.LinearGaussianStateSpaceModel
oryx.distributions.LogLogistic
oryx.distributions.LogNormal
oryx.distributions.Logistic
oryx.distributions.LogitNormal
oryx.distributions.MarkovChain
oryx.distributions.Masked
oryx.distributions.MatrixNormalLinearOperator
oryx.distributions.MatrixTLinearOperator
oryx.distributions.Mixture
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.NormalInverseGaussian
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.SigmoidBeta
oryx.distributions.SinhArcsinh
oryx.distributions.Skellam
oryx.distributions.SphericalUniform
oryx.distributions.StoppingRatioLogistic
oryx.distributions.StudentT
oryx.distributions.StudentTProcess
oryx.distributions.StudentTProcessRegressionModel
oryx.distributions.TransformedDistribution
oryx.distributions.Triangular
oryx.distributions.TruncatedCauchy
oryx.distributions.TruncatedNormal
oryx.distributions.TwoPieceNormal
oryx.distributions.Uniform
oryx.distributions.VariationalGaussianProcess
oryx.distributions.VectorDeterministic
oryx.distributions.VonMises
oryx.distributions.VonMisesFisher
oryx.distributions.Weibull
oryx.distributions.WishartLinearOperator
oryx.distributions.WishartTriL
oryx.distributions.ZeroInflatedNegativeBinomial
oryx.distributions.Zipf
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.matching
oryx.experimental.matching.jax_rewrite
oryx.experimental.matching.jax_rewrite.Bindings
oryx.experimental.matching.jax_rewrite.BoundExpression
oryx.experimental.matching.jax_rewrite.CallPrimitive
oryx.experimental.matching.jax_rewrite.Continuation
oryx.experimental.matching.jax_rewrite.Env
oryx.experimental.matching.jax_rewrite.JaxExpression
oryx.experimental.matching.jax_rewrite.JaxVar
oryx.experimental.matching.jax_rewrite.Literal
oryx.experimental.matching.jax_rewrite.Params
oryx.experimental.matching.jax_rewrite.Part
oryx.experimental.matching.jax_rewrite.Primitive
oryx.experimental.matching.jax_rewrite.Success
oryx.experimental.matching.jax_rewrite.evaluate
oryx.experimental.matching.jax_rewrite.evaluate_jax_expression
oryx.experimental.matching.jax_rewrite.evaluate_tuple
oryx.experimental.matching.jax_rewrite.evaluate_value
oryx.experimental.matching.jax_rewrite.jaxpr_to_expressions
oryx.experimental.matching.jax_rewrite.make_bound_expression
oryx.experimental.matching.jax_rewrite.matcher
oryx.experimental.matching.jax_rewrite.matcher.Choice
oryx.experimental.matching.jax_rewrite.matcher.MatchError
oryx.experimental.matching.jax_rewrite.matcher.Not
oryx.experimental.matching.jax_rewrite.matcher.Pattern
oryx.experimental.matching.jax_rewrite.matcher.Plus
oryx.experimental.matching.jax_rewrite.matcher.Segment
oryx.experimental.matching.jax_rewrite.matcher.Star
oryx.experimental.matching.jax_rewrite.matcher.Var
oryx.experimental.matching.jax_rewrite.matcher.is_match
oryx.experimental.matching.jax_rewrite.matcher.match
oryx.experimental.matching.jax_rewrite.matcher.match_all
oryx.experimental.matching.jax_rewrite.matcher.matcher
oryx.experimental.matching.jax_rewrite.primitive_to_expression
oryx.experimental.matching.jax_rewrite.rewrite
oryx.experimental.matching.jax_rewrite.rules
oryx.experimental.matching.jax_rewrite.rules.Expression
oryx.experimental.matching.jax_rewrite.rules.bottom_up
oryx.experimental.matching.jax_rewrite.rules.in_order
oryx.experimental.matching.jax_rewrite.rules.iterated
oryx.experimental.matching.jax_rewrite.rules.make_rule
oryx.experimental.matching.jax_rewrite.rules.rewrite_subexpressions
oryx.experimental.matching.jax_rewrite.rules.rule_list
oryx.experimental.matching.jax_rewrite.rules.term_rewriter
oryx.experimental.matching.jax_rewrite.rules.top_down
oryx.experimental.matching.jax_rewrite.rules.tree_children
oryx.experimental.matching.jax_rewrite.rules.tree_map
oryx.experimental.matching.jax_rewrite.trace_util
oryx.experimental.matching.jax_rewrite.trace_util.get_dynamic_context
oryx.experimental.matching.jax_rewrite.trace_util.get_shaped_aval
oryx.experimental.matching.jax_rewrite.trace_util.new_dynamic_context
oryx.experimental.matching.jax_rewrite.trace_util.pv_like
oryx.experimental.matching.jax_rewrite.trace_util.stage
oryx.experimental.matching.jax_rewrite.trace_util.trees
oryx.experimental.matching.matcher
oryx.experimental.matching.matcher.Choice
oryx.experimental.matching.matcher.MatchError
oryx.experimental.matching.matcher.Not
oryx.experimental.matching.matcher.Pattern
oryx.experimental.matching.matcher.Plus
oryx.experimental.matching.matcher.Segment
oryx.experimental.matching.matcher.Star
oryx.experimental.matching.matcher.Var
oryx.experimental.matching.matcher.is_match
oryx.experimental.matching.matcher.match
oryx.experimental.matching.matcher.match_all
oryx.experimental.matching.matcher.matcher
oryx.experimental.matching.rules
oryx.experimental.matching.rules.Expression
oryx.experimental.matching.rules.bottom_up
oryx.experimental.matching.rules.in_order
oryx.experimental.matching.rules.iterated
oryx.experimental.matching.rules.make_rule
oryx.experimental.matching.rules.rewrite_subexpressions
oryx.experimental.matching.rules.rule_list
oryx.experimental.matching.rules.term_rewriter
oryx.experimental.matching.rules.top_down
oryx.experimental.matching.rules.tree_children
oryx.experimental.matching.rules.tree_map
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