Primary symbols
oryxoryx.bijectorsoryx.bijectors.AbsoluteValueoryx.bijectors.Ascendingoryx.bijectors.AutoCompositeTensorBijectororyx.bijectors.AutoregressiveNetworkoryx.bijectors.Bijectororyx.bijectors.Blockwiseoryx.bijectors.Chainoryx.bijectors.CholeskyOuterProductoryx.bijectors.CholeskyToInvCholeskyoryx.bijectors.Compositionoryx.bijectors.CorrelationCholeskyoryx.bijectors.Cumsumoryx.bijectors.Exporyx.bijectors.Expm1oryx.bijectors.FFJORDoryx.bijectors.FillScaleTriLoryx.bijectors.FillTriangularoryx.bijectors.FrechetCDForyx.bijectors.GeneralizedExtremeValueCDForyx.bijectors.GeneralizedParetooryx.bijectors.GompertzCDForyx.bijectors.GumbelCDForyx.bijectors.Householderoryx.bijectors.Identityoryx.bijectors.Inlineoryx.bijectors.Invertoryx.bijectors.IteratedSigmoidCenteredoryx.bijectors.JointMaporyx.bijectors.KumaraswamyCDForyx.bijectors.LambertWTailoryx.bijectors.Logoryx.bijectors.Log1poryx.bijectors.MaskedAutoregressiveFloworyx.bijectors.MatrixInverseTriLoryx.bijectors.MatvecLUoryx.bijectors.MoyalCDForyx.bijectors.NormalCDForyx.bijectors.Padoryx.bijectors.Permuteoryx.bijectors.Poweroryx.bijectors.PowerTransformoryx.bijectors.RationalQuadraticSplineoryx.bijectors.RayleighCDForyx.bijectors.RealNVPoryx.bijectors.Reciprocaloryx.bijectors.Reshapeoryx.bijectors.Restructureoryx.bijectors.Scaleoryx.bijectors.ScaleMatvecDiagoryx.bijectors.ScaleMatvecLUoryx.bijectors.ScaleMatvecLinearOperatororyx.bijectors.ScaleMatvecLinearOperatorBlockoryx.bijectors.ScaleMatvecTriLoryx.bijectors.Shiftoryx.bijectors.ShiftedGompertzCDForyx.bijectors.Sigmoidoryx.bijectors.Sinhoryx.bijectors.SinhArcsinhoryx.bijectors.SoftCliporyx.bijectors.Softfloororyx.bijectors.SoftmaxCenteredoryx.bijectors.Softplusoryx.bijectors.Softsignoryx.bijectors.Splitoryx.bijectors.Squareoryx.bijectors.Tanhoryx.bijectors.TransformDiagonaloryx.bijectors.Transposeoryx.bijectors.UnitVectororyx.bijectors.WeibullCDForyx.bijectors.masked_autoregressive_default_templateoryx.bijectors.masked_denseoryx.bijectors.pack_sequence_asoryx.bijectors.real_nvp_default_templateoryx.bijectors.tree_flattenoryx.coreoryx.core.FlatPrimitiveoryx.core.HigherOrderPrimitiveoryx.core.NonInvertibleErrororyx.core.Pytreeoryx.core.call_bindoryx.core.custom_inverseoryx.core.harvestoryx.core.ildjoryx.core.interpretersoryx.core.interpreters.harvestoryx.core.interpreters.harvest.HarvestTraceoryx.core.interpreters.harvest.HarvestTraceroryx.core.interpreters.harvest.call_and_reaporyx.core.interpreters.harvest.harvestoryx.core.interpreters.harvest.nestoryx.core.interpreters.harvest.plantoryx.core.interpreters.harvest.reaporyx.core.interpreters.harvest.soworyx.core.interpreters.inverseoryx.core.interpreters.inverse.bijector_extensionsoryx.core.interpreters.inverse.bijector_extensions.patch_bijectororyx.core.interpreters.inverse.coreoryx.core.interpreters.inverse.core.InverseAndILDJoryx.core.interpreters.inverse.core.inverseoryx.core.interpreters.inverse.core.inverse_and_ildjoryx.core.interpreters.inverse.core.register_binaryoryx.core.interpreters.inverse.core.register_elementwiseoryx.core.interpreters.inverse.custom_inverseoryx.core.interpreters.inverse.custom_inverse.NonInvertibleErrororyx.core.interpreters.inverse.custom_inverse.custom_inverseoryx.core.interpreters.inverse.rulesoryx.core.interpreters.inverse.sliceoryx.core.interpreters.log_proboryx.core.interpreters.log_prob.LogProbRulesoryx.core.interpreters.log_prob.log_proboryx.core.interpreters.propagateoryx.core.interpreters.propagate.Celloryx.core.interpreters.propagate.Environmentoryx.core.interpreters.propagate.Equationoryx.core.interpreters.propagate.propagateoryx.core.inverseoryx.core.inverse_and_ildjoryx.core.log_proboryx.core.nestoryx.core.plantoryx.core.pploryx.core.ppl.LogProbFunctionoryx.core.ppl.Programoryx.core.ppl.blockoryx.core.ppl.conditionaloryx.core.ppl.effect_handleroryx.core.ppl.effect_handler.make_effect_handleroryx.core.ppl.graph_replaceoryx.core.ppl.interveneoryx.core.ppl.joint_log_proboryx.core.ppl.joint_sampleoryx.core.ppl.log_proboryx.core.ppl.make_effect_handleroryx.core.ppl.nestoryx.core.ppl.plateoryx.core.ppl.plate_utiloryx.core.ppl.plate_util.make_plateoryx.core.ppl.random_variableoryx.core.ppl.rvoryx.core.ppl.traceoryx.core.ppl.trace_log_proboryx.core.ppl.transformationsoryx.core.ppl.transformations.blockoryx.core.ppl.transformations.conditionaloryx.core.ppl.transformations.graph_replaceoryx.core.ppl.transformations.interveneoryx.core.ppl.transformations.joint_log_proboryx.core.ppl.transformations.joint_sampleoryx.core.ppl.transformations.log_proboryx.core.ppl.transformations.nestoryx.core.ppl.transformations.random_variableoryx.core.ppl.transformations.rvoryx.core.primitiveoryx.core.primitive.FlatPrimitiveoryx.core.primitive.HigherOrderPrimitiveoryx.core.primitive.call_bindoryx.core.primitive.tie_alloryx.core.primitive.tie_inoryx.core.pytreeoryx.core.pytree.Pytreeoryx.core.reaporyx.core.serializeoryx.core.serialize.deserializeoryx.core.serialize.from_tupleoryx.core.serialize.serializeoryx.core.serialize.to_tupleoryx.core.soworyx.core.stateoryx.core.state.ArraySpecoryx.core.state.Moduleoryx.core.state.Shapeoryx.core.state.apioryx.core.state.api.ArraySpecoryx.core.state.api.calloryx.core.state.api.call_and_updateoryx.core.state.api.initoryx.core.state.api.make_array_specoryx.core.state.api.specoryx.core.state.api.updateoryx.core.state.assignoryx.core.state.calloryx.core.state.call_and_updateoryx.core.state.functionoryx.core.state.function.FunctionModuleoryx.core.state.initoryx.core.state.make_array_specoryx.core.state.moduleoryx.core.state.module.Moduleoryx.core.state.module.assignoryx.core.state.module.variableoryx.core.state.specoryx.core.state.updateoryx.core.state.variableoryx.core.tie_alloryx.core.tie_inoryx.core.trace_utiloryx.core.trace_util.get_dynamic_contextoryx.core.trace_util.get_shaped_avaloryx.core.trace_util.new_dynamic_contextoryx.core.trace_util.pv_likeoryx.core.trace_util.stageoryx.core.trace_util.treesoryx.distributionsoryx.distributions.AutoCompositeTensorDistributionoryx.distributions.Autoregressiveoryx.distributions.BatchBroadcastoryx.distributions.BatchReshapeoryx.distributions.Batesoryx.distributions.Bernoullioryx.distributions.Betaoryx.distributions.BetaBinomialoryx.distributions.BetaQuotientoryx.distributions.Binomialoryx.distributions.Blockwiseoryx.distributions.Categoricaloryx.distributions.Cauchyoryx.distributions.Chioryx.distributions.Chi2oryx.distributions.CholeskyLKJoryx.distributions.DeterminantalPointProcessoryx.distributions.Deterministicoryx.distributions.Dirichletoryx.distributions.DirichletMultinomialoryx.distributions.Distributionoryx.distributions.DoublesidedMaxwelloryx.distributions.Empiricaloryx.distributions.ExpGammaoryx.distributions.ExpInverseGammaoryx.distributions.ExpRelaxedOneHotCategoricaloryx.distributions.Exponentialoryx.distributions.ExponentiallyModifiedGaussianoryx.distributions.FiniteDiscreteoryx.distributions.FlatDirichletoryx.distributions.Gammaoryx.distributions.GammaGammaoryx.distributions.GaussianProcessoryx.distributions.GaussianProcessRegressionModeloryx.distributions.GeneralizedExtremeValueoryx.distributions.GeneralizedNormaloryx.distributions.GeneralizedParetooryx.distributions.Geometricoryx.distributions.Gumbeloryx.distributions.HalfCauchyoryx.distributions.HalfNormaloryx.distributions.HalfStudentToryx.distributions.HiddenMarkovModeloryx.distributions.Horseshoeoryx.distributions.Independentoryx.distributions.Inflatedoryx.distributions.InverseGammaoryx.distributions.InverseGaussianoryx.distributions.JohnsonSUoryx.distributions.JointDistributionoryx.distributions.JointDistribution.Rootoryx.distributions.JointDistributionCoroutineoryx.distributions.JointDistributionCoroutine.Rootoryx.distributions.JointDistributionCoroutineAutoBatchedoryx.distributions.JointDistributionCoroutineAutoBatched.Rootoryx.distributions.JointDistributionNamedoryx.distributions.JointDistributionNamed.Rootoryx.distributions.JointDistributionNamedAutoBatchedoryx.distributions.JointDistributionNamedAutoBatched.Rootoryx.distributions.JointDistributionSequentialoryx.distributions.JointDistributionSequential.Rootoryx.distributions.JointDistributionSequentialAutoBatchedoryx.distributions.JointDistributionSequentialAutoBatched.Rootoryx.distributions.Kumaraswamyoryx.distributions.LKJoryx.distributions.LambertWDistributionoryx.distributions.LambertWNormaloryx.distributions.Laplaceoryx.distributions.LinearGaussianStateSpaceModeloryx.distributions.LogLogisticoryx.distributions.LogNormaloryx.distributions.Logisticoryx.distributions.LogitNormaloryx.distributions.MarkovChainoryx.distributions.Maskedoryx.distributions.MatrixNormalLinearOperatororyx.distributions.MatrixTLinearOperatororyx.distributions.Mixtureoryx.distributions.MixtureSameFamilyoryx.distributions.Moyaloryx.distributions.Multinomialoryx.distributions.MultivariateNormalDiagoryx.distributions.MultivariateNormalDiagPlusLowRankoryx.distributions.MultivariateNormalDiagPlusLowRankCovarianceoryx.distributions.MultivariateNormalFullCovarianceoryx.distributions.MultivariateNormalLinearOperatororyx.distributions.MultivariateNormalTriLoryx.distributions.MultivariateStudentTLinearOperatororyx.distributions.NegativeBinomialoryx.distributions.NoncentralChi2oryx.distributions.Normaloryx.distributions.NormalInverseGaussianoryx.distributions.OneHotCategoricaloryx.distributions.OrderedLogisticoryx.distributions.PERToryx.distributions.Paretooryx.distributions.PlackettLuceoryx.distributions.Poissonoryx.distributions.PoissonLogNormalQuadratureCompoundoryx.distributions.PowerSphericaloryx.distributions.ProbitBernoullioryx.distributions.QuantizedDistributionoryx.distributions.RegisterKLoryx.distributions.RelaxedBernoullioryx.distributions.RelaxedOneHotCategoricaloryx.distributions.ReparameterizationTypeoryx.distributions.Sampleoryx.distributions.SigmoidBetaoryx.distributions.SinhArcsinhoryx.distributions.Skellamoryx.distributions.SphericalUniformoryx.distributions.StoppingRatioLogisticoryx.distributions.StudentToryx.distributions.StudentTProcessoryx.distributions.StudentTProcessRegressionModeloryx.distributions.TransformedDistributionoryx.distributions.Triangularoryx.distributions.TruncatedCauchyoryx.distributions.TruncatedNormaloryx.distributions.TwoPieceNormaloryx.distributions.TwoPieceStudentToryx.distributions.Uniformoryx.distributions.VariationalGaussianProcessoryx.distributions.VectorDeterministicoryx.distributions.VonMisesoryx.distributions.VonMisesFisheroryx.distributions.Weibulloryx.distributions.WishartLinearOperatororyx.distributions.WishartTriLoryx.distributions.ZeroInflatedNegativeBinomialoryx.distributions.Zipforyx.distributions.independent_joint_distribution_from_structureoryx.distributions.kl_divergenceoryx.distributions.mvn_conjugate_linear_updateoryx.distributions.normal_conjugates_known_scale_posteriororyx.distributions.normal_conjugates_known_scale_predictiveoryx.distributions.quadrature_scheme_lognormal_gauss_hermiteoryx.distributions.quadrature_scheme_lognormal_quantilesoryx.experimentaloryx.experimental.matchingoryx.experimental.matching.jax_rewriteoryx.experimental.matching.jax_rewrite.Bindingsoryx.experimental.matching.jax_rewrite.BoundExpressionoryx.experimental.matching.jax_rewrite.CallPrimitiveoryx.experimental.matching.jax_rewrite.Continuationoryx.experimental.matching.jax_rewrite.Envoryx.experimental.matching.jax_rewrite.JaxExpressionoryx.experimental.matching.jax_rewrite.JaxVaroryx.experimental.matching.jax_rewrite.Literaloryx.experimental.matching.jax_rewrite.Paramsoryx.experimental.matching.jax_rewrite.Partoryx.experimental.matching.jax_rewrite.PjitPrimitiveoryx.experimental.matching.jax_rewrite.Primitiveoryx.experimental.matching.jax_rewrite.Successoryx.experimental.matching.jax_rewrite.evaluateoryx.experimental.matching.jax_rewrite.evaluate_jax_expressionoryx.experimental.matching.jax_rewrite.evaluate_tupleoryx.experimental.matching.jax_rewrite.evaluate_valueoryx.experimental.matching.jax_rewrite.jaxpr_to_expressionsoryx.experimental.matching.jax_rewrite.make_bound_expressionoryx.experimental.matching.jax_rewrite.matcheroryx.experimental.matching.jax_rewrite.matcher.Choiceoryx.experimental.matching.jax_rewrite.matcher.MatchErrororyx.experimental.matching.jax_rewrite.matcher.Notoryx.experimental.matching.jax_rewrite.matcher.Patternoryx.experimental.matching.jax_rewrite.matcher.Plusoryx.experimental.matching.jax_rewrite.matcher.Segmentoryx.experimental.matching.jax_rewrite.matcher.Staroryx.experimental.matching.jax_rewrite.matcher.Varoryx.experimental.matching.jax_rewrite.matcher.is_matchoryx.experimental.matching.jax_rewrite.matcher.matchoryx.experimental.matching.jax_rewrite.matcher.match_alloryx.experimental.matching.jax_rewrite.matcher.matcheroryx.experimental.matching.jax_rewrite.primitive_to_expressionoryx.experimental.matching.jax_rewrite.rewriteoryx.experimental.matching.jax_rewrite.rulesoryx.experimental.matching.jax_rewrite.rules.Expressionoryx.experimental.matching.jax_rewrite.rules.bottom_uporyx.experimental.matching.jax_rewrite.rules.in_orderoryx.experimental.matching.jax_rewrite.rules.iteratedoryx.experimental.matching.jax_rewrite.rules.make_ruleoryx.experimental.matching.jax_rewrite.rules.rewrite_subexpressionsoryx.experimental.matching.jax_rewrite.rules.rule_listoryx.experimental.matching.jax_rewrite.rules.term_rewriteroryx.experimental.matching.jax_rewrite.rules.top_downoryx.experimental.matching.jax_rewrite.rules.tree_childrenoryx.experimental.matching.jax_rewrite.rules.tree_maporyx.experimental.matching.jax_rewrite.trace_utiloryx.experimental.matching.jax_rewrite.trace_util.get_dynamic_contextoryx.experimental.matching.jax_rewrite.trace_util.get_shaped_avaloryx.experimental.matching.jax_rewrite.trace_util.new_dynamic_contextoryx.experimental.matching.jax_rewrite.trace_util.pv_likeoryx.experimental.matching.jax_rewrite.trace_util.stageoryx.experimental.matching.jax_rewrite.trace_util.treesoryx.experimental.matching.matcheroryx.experimental.matching.matcher.Choiceoryx.experimental.matching.matcher.MatchErrororyx.experimental.matching.matcher.Notoryx.experimental.matching.matcher.Patternoryx.experimental.matching.matcher.Plusoryx.experimental.matching.matcher.Segmentoryx.experimental.matching.matcher.Staroryx.experimental.matching.matcher.Varoryx.experimental.matching.matcher.is_matchoryx.experimental.matching.matcher.matchoryx.experimental.matching.matcher.match_alloryx.experimental.matching.matcher.matcheroryx.experimental.matching.rulesoryx.experimental.matching.rules.Expressionoryx.experimental.matching.rules.bottom_uporyx.experimental.matching.rules.in_orderoryx.experimental.matching.rules.iteratedoryx.experimental.matching.rules.make_ruleoryx.experimental.matching.rules.rewrite_subexpressionsoryx.experimental.matching.rules.rule_listoryx.experimental.matching.rules.term_rewriteroryx.experimental.matching.rules.top_downoryx.experimental.matching.rules.tree_childrenoryx.experimental.matching.rules.tree_maporyx.experimental.mcmcoryx.experimental.mcmc.constrainoryx.experimental.mcmc.hmcoryx.experimental.mcmc.kernelsoryx.experimental.mcmc.kernels.hmcoryx.experimental.mcmc.kernels.malaoryx.experimental.mcmc.kernels.metropolisoryx.experimental.mcmc.kernels.metropolis_hastingsoryx.experimental.mcmc.kernels.random_walkoryx.experimental.mcmc.kernels.sample_chainoryx.experimental.mcmc.malaoryx.experimental.mcmc.metropolisoryx.experimental.mcmc.metropolis_hastingsoryx.experimental.mcmc.random_walkoryx.experimental.mcmc.sample_chainoryx.experimental.mcmc.utilsoryx.experimental.mcmc.utils.constrainoryx.experimental.nnoryx.experimental.nn.AvgPoolingoryx.experimental.nn.BatchNormoryx.experimental.nn.Convoryx.experimental.nn.Deconvoryx.experimental.nn.Denseoryx.experimental.nn.Dropoutoryx.experimental.nn.Flattenoryx.experimental.nn.Layeroryx.experimental.nn.LayerParamsoryx.experimental.nn.LogSoftmaxoryx.experimental.nn.MaxPoolingoryx.experimental.nn.Reluoryx.experimental.nn.Reshapeoryx.experimental.nn.Serialoryx.experimental.nn.Softmaxoryx.experimental.nn.Softplusoryx.experimental.nn.SumPoolingoryx.experimental.nn.Tanhoryx.experimental.nn.Templateoryx.experimental.nn.baseoryx.experimental.nn.base.Layeroryx.experimental.nn.base.LayerParamsoryx.experimental.nn.base.Templateoryx.experimental.nn.base.create_parameteroryx.experimental.nn.combinatororyx.experimental.nn.combinator.Serialoryx.experimental.nn.convolutionoryx.experimental.nn.convolution.Convoryx.experimental.nn.convolution.Deconvoryx.experimental.nn.convolution.conv_infooryx.experimental.nn.coreoryx.experimental.nn.core.Activationoryx.experimental.nn.core.Denseoryx.experimental.nn.core.Dropoutoryx.experimental.nn.core.LogSoftmaxoryx.experimental.nn.core.Reluoryx.experimental.nn.core.Softmaxoryx.experimental.nn.core.Softplusoryx.experimental.nn.core.Tanhoryx.experimental.nn.normalizationoryx.experimental.nn.normalization.BatchNormoryx.experimental.nn.poolingoryx.experimental.nn.pooling.AvgPoolingoryx.experimental.nn.pooling.MaxPoolingoryx.experimental.nn.pooling.Poolingoryx.experimental.nn.pooling.SumPoolingoryx.experimental.nn.reshapeoryx.experimental.nn.reshape.Flattenoryx.experimental.nn.reshape.Reshapeoryx.experimental.optimizersoryx.experimental.optimizers.adamoryx.experimental.optimizers.gradient_descentoryx.experimental.optimizers.noisy_sgdoryx.experimental.optimizers.optimizeoryx.experimental.optimizers.optixoryx.experimental.optimizers.optix.adamoryx.experimental.optimizers.optix.add_noiseoryx.experimental.optimizers.optix.apply_everyoryx.experimental.optimizers.optix.chainoryx.experimental.optimizers.optix.cliporyx.experimental.optimizers.optix.clip_by_global_normoryx.experimental.optimizers.optix.global_normoryx.experimental.optimizers.optix.gradient_descentoryx.experimental.optimizers.optix.noisy_sgdoryx.experimental.optimizers.optix.optimizeoryx.experimental.optimizers.optix.rmsproporyx.experimental.optimizers.optix.scaleoryx.experimental.optimizers.optix.scale_by_adamoryx.experimental.optimizers.optix.scale_by_rmsoryx.experimental.optimizers.optix.scale_by_scheduleoryx.experimental.optimizers.optix.scale_by_stddevoryx.experimental.optimizers.optix.sgdoryx.experimental.optimizers.optix.traceoryx.experimental.optimizers.rmsproporyx.experimental.optimizers.sgdoryx.internaloryx.internal.test_utiloryx.internal.test_util.TestCaseoryx.internal.test_util.TestCase.failureExceptionoryx.utiloryx.util.get_summariesoryx.util.summary