Primary symbols
tfptfp.bijectorstfp.bijectors.AbsoluteValuetfp.bijectors.Affinetfp.bijectors.AffineLinearOperatortfp.bijectors.AffineScalartfp.bijectors.Ascendingtfp.bijectors.AutoCompositeTensorBijectortfp.bijectors.AutoregressiveNetworktfp.bijectors.BatchNormalizationtfp.bijectors.Bijectortfp.bijectors.Blockwisetfp.bijectors.Chaintfp.bijectors.CholeskyOuterProducttfp.bijectors.CholeskyToInvCholeskytfp.bijectors.Compositiontfp.bijectors.CorrelationCholeskytfp.bijectors.Cumsumtfp.bijectors.DiscreteCosineTransformtfp.bijectors.Exptfp.bijectors.Expm1tfp.bijectors.FFJORDtfp.bijectors.FillScaleTriLtfp.bijectors.FillTriangulartfp.bijectors.FrechetCDFtfp.bijectors.GeneralizedExtremeValueCDFtfp.bijectors.GeneralizedParetotfp.bijectors.Glowtfp.bijectors.GlowDefaultExitNetworktfp.bijectors.GlowDefaultNetworktfp.bijectors.GompertzCDFtfp.bijectors.GumbelCDFtfp.bijectors.Identitytfp.bijectors.Inlinetfp.bijectors.Inverttfp.bijectors.IteratedSigmoidCenteredtfp.bijectors.JointMaptfp.bijectors.KumaraswamyCDFtfp.bijectors.LambertWTailtfp.bijectors.Logtfp.bijectors.Log1ptfp.bijectors.MaskedAutoregressiveFlowtfp.bijectors.MatrixInverseTriLtfp.bijectors.MatvecLUtfp.bijectors.MoyalCDFtfp.bijectors.NormalCDFtfp.bijectors.Orderedtfp.bijectors.Padtfp.bijectors.Permutetfp.bijectors.Powertfp.bijectors.PowerTransformtfp.bijectors.RationalQuadraticSplinetfp.bijectors.RayleighCDFtfp.bijectors.RealNVPtfp.bijectors.Reciprocaltfp.bijectors.Reshapetfp.bijectors.Restructuretfp.bijectors.Scaletfp.bijectors.ScaleMatvecDiagtfp.bijectors.ScaleMatvecLUtfp.bijectors.ScaleMatvecLinearOperatortfp.bijectors.ScaleMatvecLinearOperatorBlocktfp.bijectors.ScaleMatvecTriLtfp.bijectors.Shifttfp.bijectors.ShiftedGompertzCDFtfp.bijectors.Sigmoidtfp.bijectors.Sinhtfp.bijectors.SinhArcsinhtfp.bijectors.SoftCliptfp.bijectors.Softfloortfp.bijectors.SoftmaxCenteredtfp.bijectors.Softplustfp.bijectors.Softsigntfp.bijectors.Splittfp.bijectors.Squaretfp.bijectors.Tanhtfp.bijectors.TransformDiagonaltfp.bijectors.Transposetfp.bijectors.WeibullCDFtfp.bijectors.masked_autoregressive_default_templatetfp.bijectors.masked_densetfp.bijectors.pack_sequence_astfp.bijectors.real_nvp_default_templatetfp.bijectors.tree_flattentfp.debuggingtfp.debugging.benchmarkingtfp.debugging.benchmarking.BenchmarkTfFunctionConfigtfp.debugging.benchmarking.benchmark_tf_functiontfp.debugging.benchmarking.default_benchmark_configtfp.distributionstfp.distributions.Autoregressivetfp.distributions.BatchBroadcasttfp.distributions.BatchReshapetfp.distributions.Batestfp.distributions.Bernoullitfp.distributions.Betatfp.distributions.BetaBinomialtfp.distributions.BetaQuotienttfp.distributions.Binomialtfp.distributions.Blockwisetfp.distributions.Categoricaltfp.distributions.Cauchytfp.distributions.Chitfp.distributions.Chi2tfp.distributions.CholeskyLKJtfp.distributions.DeterminantalPointProcesstfp.distributions.Deterministictfp.distributions.Dirichlettfp.distributions.DirichletMultinomialtfp.distributions.Distributiontfp.distributions.DoublesidedMaxwelltfp.distributions.Empiricaltfp.distributions.ExpGammatfp.distributions.ExpInverseGammatfp.distributions.ExpRelaxedOneHotCategoricaltfp.distributions.Exponentialtfp.distributions.ExponentiallyModifiedGaussiantfp.distributions.Gammatfp.distributions.GammaGammatfp.distributions.GaussianProcesstfp.distributions.GaussianProcessRegressionModeltfp.distributions.GeneralizedExtremeValuetfp.distributions.GeneralizedNormaltfp.distributions.GeneralizedParetotfp.distributions.Geometrictfp.distributions.Gumbeltfp.distributions.HalfCauchytfp.distributions.HalfNormaltfp.distributions.HalfStudentTtfp.distributions.HiddenMarkovModeltfp.distributions.Horseshoetfp.distributions.Independenttfp.distributions.InverseGammatfp.distributions.InverseGaussiantfp.distributions.JohnsonSUtfp.distributions.JointDistributiontfp.distributions.JointDistribution.Roottfp.distributions.JointDistributionCoroutinetfp.distributions.JointDistributionCoroutine.Roottfp.distributions.JointDistributionCoroutineAutoBatchedtfp.distributions.JointDistributionCoroutineAutoBatched.Roottfp.distributions.JointDistributionNamedtfp.distributions.JointDistributionNamed.Roottfp.distributions.JointDistributionNamedAutoBatchedtfp.distributions.JointDistributionNamedAutoBatched.Roottfp.distributions.JointDistributionSequentialtfp.distributions.JointDistributionSequential.Roottfp.distributions.JointDistributionSequentialAutoBatchedtfp.distributions.JointDistributionSequentialAutoBatched.Roottfp.distributions.Kumaraswamytfp.distributions.LKJtfp.distributions.LambertWDistributiontfp.distributions.LambertWNormaltfp.distributions.Laplacetfp.distributions.LinearGaussianStateSpaceModeltfp.distributions.LogLogistictfp.distributions.LogNormaltfp.distributions.Logistictfp.distributions.LogitNormaltfp.distributions.Maskedtfp.distributions.MatrixNormalLinearOperatortfp.distributions.MatrixTLinearOperatortfp.distributions.Mixturetfp.distributions.MixtureSameFamilytfp.distributions.Moyaltfp.distributions.Multinomialtfp.distributions.MultivariateNormalDiagtfp.distributions.MultivariateNormalDiagPlusLowRanktfp.distributions.MultivariateNormalFullCovariancetfp.distributions.MultivariateNormalLinearOperatortfp.distributions.MultivariateNormalTriLtfp.distributions.MultivariateStudentTLinearOperatortfp.distributions.NegativeBinomialtfp.distributions.Normaltfp.distributions.NormalInverseGaussiantfp.distributions.OneHotCategoricaltfp.distributions.OrderedLogistictfp.distributions.PERTtfp.distributions.Paretotfp.distributions.PixelCNNtfp.distributions.PlackettLucetfp.distributions.Poissontfp.distributions.PoissonLogNormalQuadratureCompoundtfp.distributions.PowerSphericaltfp.distributions.ProbitBernoullitfp.distributions.QuantizedDistributiontfp.distributions.RegisterKLtfp.distributions.RelaxedBernoullitfp.distributions.RelaxedOneHotCategoricaltfp.distributions.ReparameterizationTypetfp.distributions.Sampletfp.distributions.SigmoidBetatfp.distributions.SinhArcsinhtfp.distributions.Skellamtfp.distributions.SphericalUniformtfp.distributions.StoppingRatioLogistictfp.distributions.StudentTtfp.distributions.StudentTProcesstfp.distributions.TransformedDistributiontfp.distributions.Triangulartfp.distributions.TruncatedCauchytfp.distributions.TruncatedNormaltfp.distributions.Uniformtfp.distributions.VariationalGaussianProcesstfp.distributions.VectorDeterministictfp.distributions.VectorExponentialDiagtfp.distributions.VonMisestfp.distributions.VonMisesFishertfp.distributions.Weibulltfp.distributions.WishartLinearOperatortfp.distributions.WishartTriLtfp.distributions.Zipftfp.distributions.independent_joint_distribution_from_structuretfp.distributions.kl_divergencetfp.distributions.mvn_conjugate_linear_updatetfp.distributions.normal_conjugates_known_scale_posteriortfp.distributions.normal_conjugates_known_scale_predictivetfp.distributions.quadrature_scheme_lognormal_gauss_hermitetfp.distributions.quadrature_scheme_lognormal_quantilestfp.experimentaltfp.experimental.AutoCompositeTensortfp.experimental.as_compositetfp.experimental.auto_batchingtfp.experimental.auto_batching.Contexttfp.experimental.auto_batching.NumpyBackendtfp.experimental.auto_batching.TensorFlowBackendtfp.experimental.auto_batching.TensorTypetfp.experimental.auto_batching.Typetfp.experimental.auto_batching.allocation_strategytfp.experimental.auto_batching.allocation_strategy.optimizetfp.experimental.auto_batching.dsltfp.experimental.auto_batching.dsl.ProgramBuildertfp.experimental.auto_batching.frontendtfp.experimental.auto_batching.frontend.Contexttfp.experimental.auto_batching.frontend.ab_type_inferencetfp.experimental.auto_batching.frontend.ab_type_inference.infer_typestfp.experimental.auto_batching.frontend.ab_type_inference.infer_types_from_signaturetfp.experimental.auto_batching.frontend.ab_type_inference.is_inferringtfp.experimental.auto_batching.frontend.ab_type_inference.signaturetfp.experimental.auto_batching.frontend.ab_type_inference.type_of_patterntfp.experimental.auto_batching.frontend.allocation_strategytfp.experimental.auto_batching.frontend.allocation_strategy.optimizetfp.experimental.auto_batching.frontend.dsltfp.experimental.auto_batching.frontend.dsl.ProgramBuildertfp.experimental.auto_batching.frontend.gast_utiltfp.experimental.auto_batching.frontend.gast_util.Moduletfp.experimental.auto_batching.frontend.gast_util.Nametfp.experimental.auto_batching.frontend.gast_util.Strtfp.experimental.auto_batching.frontend.gast_util.is_constanttfp.experimental.auto_batching.frontend.gast_util.is_ellipsistfp.experimental.auto_batching.frontend.gast_util.is_literaltfp.experimental.auto_batching.frontend.instructionstfp.experimental.auto_batching.frontend.instructions.Blocktfp.experimental.auto_batching.frontend.instructions.BranchOptfp.experimental.auto_batching.frontend.instructions.ControlFlowGraphtfp.experimental.auto_batching.frontend.instructions.Functiontfp.experimental.auto_batching.frontend.instructions.FunctionCallOptfp.experimental.auto_batching.frontend.instructions.GotoOptfp.experimental.auto_batching.frontend.instructions.IndirectGotoOptfp.experimental.auto_batching.frontend.instructions.PopOptfp.experimental.auto_batching.frontend.instructions.PrimOptfp.experimental.auto_batching.frontend.instructions.Programtfp.experimental.auto_batching.frontend.instructions.PushGotoOptfp.experimental.auto_batching.frontend.instructions.TensorTypetfp.experimental.auto_batching.frontend.instructions.Typetfp.experimental.auto_batching.frontend.instructions.VariableAllocationtfp.experimental.auto_batching.frontend.instructions.extract_referenced_variablestfp.experimental.auto_batching.frontend.instructions.halt_optfp.experimental.auto_batching.frontend.instructions.interprettfp.experimental.auto_batching.frontend.instructions.push_optfp.experimental.auto_batching.frontend.loweringtfp.experimental.auto_batching.frontend.lowering.lower_function_callstfp.experimental.auto_batching.frontend.sttfp.experimental.auto_batching.frontend.st.ExecutionQueuetfp.experimental.auto_batching.frontend.st.executetfp.experimental.auto_batching.frontend.st.insttfp.experimental.auto_batching.frontend.st.inst.Blocktfp.experimental.auto_batching.frontend.st.inst.BranchOptfp.experimental.auto_batching.frontend.st.inst.ControlFlowGraphtfp.experimental.auto_batching.frontend.st.inst.Functiontfp.experimental.auto_batching.frontend.st.inst.FunctionCallOptfp.experimental.auto_batching.frontend.st.inst.GotoOptfp.experimental.auto_batching.frontend.st.inst.IndirectGotoOptfp.experimental.auto_batching.frontend.st.inst.PopOptfp.experimental.auto_batching.frontend.st.inst.PrimOptfp.experimental.auto_batching.frontend.st.inst.Programtfp.experimental.auto_batching.frontend.st.inst.PushGotoOptfp.experimental.auto_batching.frontend.st.inst.TensorTypetfp.experimental.auto_batching.frontend.st.inst.Typetfp.experimental.auto_batching.frontend.st.inst.VariableAllocationtfp.experimental.auto_batching.frontend.st.inst.extract_referenced_variablestfp.experimental.auto_batching.frontend.st.inst.halt_optfp.experimental.auto_batching.frontend.st.inst.interprettfp.experimental.auto_batching.frontend.st.inst.push_optfp.experimental.auto_batching.frontend.st.is_runningtfp.experimental.auto_batching.frontend.stacktfp.experimental.auto_batching.frontend.stack.fuse_pop_pushtfp.experimental.auto_batching.frontend.stack.insttfp.experimental.auto_batching.frontend.stack.inst.Blocktfp.experimental.auto_batching.frontend.stack.inst.BranchOptfp.experimental.auto_batching.frontend.stack.inst.ControlFlowGraphtfp.experimental.auto_batching.frontend.stack.inst.Functiontfp.experimental.auto_batching.frontend.stack.inst.FunctionCallOptfp.experimental.auto_batching.frontend.stack.inst.GotoOptfp.experimental.auto_batching.frontend.stack.inst.IndirectGotoOptfp.experimental.auto_batching.frontend.stack.inst.PopOptfp.experimental.auto_batching.frontend.stack.inst.PrimOptfp.experimental.auto_batching.frontend.stack.inst.Programtfp.experimental.auto_batching.frontend.stack.inst.PushGotoOptfp.experimental.auto_batching.frontend.stack.inst.TensorTypetfp.experimental.auto_batching.frontend.stack.inst.Typetfp.experimental.auto_batching.frontend.stack.inst.VariableAllocationtfp.experimental.auto_batching.frontend.stack.inst.extract_referenced_variablestfp.experimental.auto_batching.frontend.stack.inst.halt_optfp.experimental.auto_batching.frontend.stack.inst.interprettfp.experimental.auto_batching.frontend.stack.inst.push_optfp.experimental.auto_batching.frontend.tf_backendtfp.experimental.auto_batching.frontend.tf_backend.TensorFlowBackendtfp.experimental.auto_batching.frontend.truthytfp.experimental.auto_batching.frontend.vmtfp.experimental.auto_batching.frontend.vm.executetfp.experimental.auto_batching.frontend.vm.is_stagingtfp.experimental.auto_batching.instructionstfp.experimental.auto_batching.instructions.Blocktfp.experimental.auto_batching.instructions.BranchOptfp.experimental.auto_batching.instructions.ControlFlowGraphtfp.experimental.auto_batching.instructions.Functiontfp.experimental.auto_batching.instructions.FunctionCallOptfp.experimental.auto_batching.instructions.GotoOptfp.experimental.auto_batching.instructions.IndirectGotoOptfp.experimental.auto_batching.instructions.PopOptfp.experimental.auto_batching.instructions.PrimOptfp.experimental.auto_batching.instructions.Programtfp.experimental.auto_batching.instructions.PushGotoOptfp.experimental.auto_batching.instructions.TensorTypetfp.experimental.auto_batching.instructions.Typetfp.experimental.auto_batching.instructions.VariableAllocationtfp.experimental.auto_batching.instructions.extract_referenced_variablestfp.experimental.auto_batching.instructions.halt_optfp.experimental.auto_batching.instructions.interprettfp.experimental.auto_batching.instructions.push_optfp.experimental.auto_batching.livenesstfp.experimental.auto_batching.liveness.liveness_analysistfp.experimental.auto_batching.loweringtfp.experimental.auto_batching.lowering.lower_function_callstfp.experimental.auto_batching.numpy_backendtfp.experimental.auto_batching.numpy_backend.NumpyBackendtfp.experimental.auto_batching.stack_optimizationtfp.experimental.auto_batching.stack_optimization.fuse_pop_pushtfp.experimental.auto_batching.stack_optimization.insttfp.experimental.auto_batching.stack_optimization.inst.Blocktfp.experimental.auto_batching.stack_optimization.inst.BranchOptfp.experimental.auto_batching.stack_optimization.inst.ControlFlowGraphtfp.experimental.auto_batching.stack_optimization.inst.Functiontfp.experimental.auto_batching.stack_optimization.inst.FunctionCallOptfp.experimental.auto_batching.stack_optimization.inst.GotoOptfp.experimental.auto_batching.stack_optimization.inst.IndirectGotoOptfp.experimental.auto_batching.stack_optimization.inst.PopOptfp.experimental.auto_batching.stack_optimization.inst.PrimOptfp.experimental.auto_batching.stack_optimization.inst.Programtfp.experimental.auto_batching.stack_optimization.inst.PushGotoOptfp.experimental.auto_batching.stack_optimization.inst.TensorTypetfp.experimental.auto_batching.stack_optimization.inst.Typetfp.experimental.auto_batching.stack_optimization.inst.VariableAllocationtfp.experimental.auto_batching.stack_optimization.inst.extract_referenced_variablestfp.experimental.auto_batching.stack_optimization.inst.halt_optfp.experimental.auto_batching.stack_optimization.inst.interprettfp.experimental.auto_batching.stack_optimization.inst.push_optfp.experimental.auto_batching.stacklesstfp.experimental.auto_batching.stackless.ExecutionQueuetfp.experimental.auto_batching.stackless.executetfp.experimental.auto_batching.stackless.insttfp.experimental.auto_batching.stackless.inst.Blocktfp.experimental.auto_batching.stackless.inst.BranchOptfp.experimental.auto_batching.stackless.inst.ControlFlowGraphtfp.experimental.auto_batching.stackless.inst.Functiontfp.experimental.auto_batching.stackless.inst.FunctionCallOptfp.experimental.auto_batching.stackless.inst.GotoOptfp.experimental.auto_batching.stackless.inst.IndirectGotoOptfp.experimental.auto_batching.stackless.inst.PopOptfp.experimental.auto_batching.stackless.inst.PrimOptfp.experimental.auto_batching.stackless.inst.Programtfp.experimental.auto_batching.stackless.inst.PushGotoOptfp.experimental.auto_batching.stackless.inst.TensorTypetfp.experimental.auto_batching.stackless.inst.Typetfp.experimental.auto_batching.stackless.inst.VariableAllocationtfp.experimental.auto_batching.stackless.inst.extract_referenced_variablestfp.experimental.auto_batching.stackless.inst.halt_optfp.experimental.auto_batching.stackless.inst.interprettfp.experimental.auto_batching.stackless.inst.push_optfp.experimental.auto_batching.stackless.is_runningtfp.experimental.auto_batching.tf_backendtfp.experimental.auto_batching.tf_backend.TensorFlowBackendtfp.experimental.auto_batching.truthytfp.experimental.auto_batching.type_inferencetfp.experimental.auto_batching.type_inference.infer_typestfp.experimental.auto_batching.type_inference.infer_types_from_signaturetfp.experimental.auto_batching.type_inference.is_inferringtfp.experimental.auto_batching.type_inference.signaturetfp.experimental.auto_batching.type_inference.type_of_patterntfp.experimental.auto_batching.virtual_machinetfp.experimental.auto_batching.virtual_machine.executetfp.experimental.auto_batching.virtual_machine.is_stagingtfp.experimental.auto_batching.xlatfp.experimental.auto_batching.xla.compile_nested_outputtfp.experimental.auto_composite_tensortfp.experimental.bijectorstfp.experimental.bijectors.ScalarFunctionWithInferredInversetfp.experimental.bijectors.Shardedtfp.experimental.bijectors.forward_log_det_jacobian_ratiotfp.experimental.bijectors.inverse_log_det_jacobian_ratiotfp.experimental.bijectors.make_distribution_bijectortfp.experimental.distributetfp.experimental.distribute.JointDistributionCoroutinetfp.experimental.distribute.JointDistributionCoroutine.Roottfp.experimental.distribute.JointDistributionNamedtfp.experimental.distribute.JointDistributionNamed.Roottfp.experimental.distribute.JointDistributionSequentialtfp.experimental.distribute.JointDistributionSequential.Roottfp.experimental.distribute.Shardedtfp.experimental.distribute.make_sharded_log_prob_partstfp.experimental.distributionstfp.experimental.distributions.IncrementLogProbtfp.experimental.distributions.JointDensityCoroutinetfp.experimental.distributions.JointDistributionPinnedtfp.experimental.distributions.MultivariateNormalPrecisionFactorLinearOperatortfp.experimental.distributions.log_prob_ratiotfp.experimental.distributions.marginal_fnstfp.experimental.distributions.marginal_fns.make_backoff_choleksytfp.experimental.distributions.marginal_fns.make_cholesky_like_marginal_fntfp.experimental.distributions.marginal_fns.make_eigh_marginal_fntfp.experimental.distributions.marginal_fns.mvn_linear_operatortfp.experimental.distributions.marginal_fns.mvn_linear_operator.MultivariateNormalLinearOperatortfp.experimental.lazybonestfp.experimental.lazybones.Deferredtfp.experimental.lazybones.DeferredBasetfp.experimental.lazybones.DeferredInputtfp.experimental.lazybones.DeferredScopetfp.experimental.lazybones.utilstfp.experimental.lazybones.utils.HashableWeakReftfp.experimental.lazybones.utils.ObjectProxytfp.experimental.lazybones.utils.SpecialMethodstfp.experimental.lazybones.utils.WeakKeyDictionarytfp.experimental.lazybones.utils.WeakSettfp.experimental.lazybones.utils.distribution_measuretfp.experimental.lazybones.utils.get_leavestfp.experimental.lazybones.utils.get_rootstfp.experimental.lazybones.utils.is_any_ancestortfp.experimental.lazybones.utils.iter_edgestfp.experimental.lazybones.utils.log_probtfp.experimental.lazybones.utils.plot_graphtfp.experimental.lazybones.utils.probtfp.experimental.linalgtfp.experimental.linalg.LinearOperatorPSDKerneltfp.experimental.linalg.no_pivot_ldltfp.experimental.linalg.simple_robustified_choleskytfp.experimental.marginalizetfp.experimental.marginalize.MarginalizableJointDistributionCoroutinetfp.experimental.marginalize.MarginalizableJointDistributionCoroutine.Roottfp.experimental.marginalize.logeinsumexptfp.experimental.mcmctfp.experimental.mcmc.CovarianceReducertfp.experimental.mcmc.DiagonalMassMatrixAdaptationtfp.experimental.mcmc.EllipticalSliceSamplertfp.experimental.mcmc.ExpectationsReducertfp.experimental.mcmc.GradientBasedTrajectoryLengthAdaptationtfp.experimental.mcmc.GradientBasedTrajectoryLengthAdaptationResultstfp.experimental.mcmc.KernelBuildertfp.experimental.mcmc.KernelOutputstfp.experimental.mcmc.NoUTurnSamplertfp.experimental.mcmc.PotentialScaleReductionReducertfp.experimental.mcmc.PreconditionedHamiltonianMonteCarlotfp.experimental.mcmc.PreconditionedNoUTurnSamplertfp.experimental.mcmc.ProgressBarReducertfp.experimental.mcmc.Reducertfp.experimental.mcmc.SampleDiscardingKerneltfp.experimental.mcmc.SequentialMonteCarlotfp.experimental.mcmc.SequentialMonteCarloResultstfp.experimental.mcmc.StateWithHistorytfp.experimental.mcmc.ThinningKerneltfp.experimental.mcmc.TracingReducertfp.experimental.mcmc.VarianceReducertfp.experimental.mcmc.WeightedParticlestfp.experimental.mcmc.WithReductionstfp.experimental.mcmc.WithReductionsKernelResultstfp.experimental.mcmc.augment_prior_with_state_historytfp.experimental.mcmc.augment_with_observation_historytfp.experimental.mcmc.augment_with_state_historytfp.experimental.mcmc.chees_criteriontfp.experimental.mcmc.default_make_hmc_kernel_fntfp.experimental.mcmc.ess_below_thresholdtfp.experimental.mcmc.gen_make_hmc_kernel_fntfp.experimental.mcmc.gen_make_transform_hmc_kernel_fntfp.experimental.mcmc.infer_trajectoriestfp.experimental.mcmc.init_near_unconstrained_zerotfp.experimental.mcmc.make_rwmh_kernel_fntfp.experimental.mcmc.make_tqdm_progress_bar_fntfp.experimental.mcmc.particle_filtertfp.experimental.mcmc.reconstruct_trajectoriestfp.experimental.mcmc.remc_thermodynamic_integralstfp.experimental.mcmc.resample_deterministic_minimum_errortfp.experimental.mcmc.resample_independenttfp.experimental.mcmc.resample_stratifiedtfp.experimental.mcmc.resample_systematictfp.experimental.mcmc.retry_inittfp.experimental.mcmc.sample_chaintfp.experimental.mcmc.sample_chain_with_burnintfp.experimental.mcmc.sample_foldtfp.experimental.mcmc.sample_sequential_monte_carlotfp.experimental.mcmc.simple_heuristic_tuningtfp.experimental.mcmc.step_kerneltfp.experimental.mcmc.windowed_adaptive_hmctfp.experimental.mcmc.windowed_adaptive_nutstfp.experimental.nntfp.experimental.nn.Affinetfp.experimental.nn.AffineVariationalFlipouttfp.experimental.nn.AffineVariationalReparameterizationtfp.experimental.nn.AffineVariationalReparameterizationLocaltfp.experimental.nn.Convolutiontfp.experimental.nn.ConvolutionTransposetfp.experimental.nn.ConvolutionTransposeVariationalFlipouttfp.experimental.nn.ConvolutionTransposeVariationalReparameterizationtfp.experimental.nn.ConvolutionV2tfp.experimental.nn.ConvolutionVariationalFlipouttfp.experimental.nn.ConvolutionVariationalFlipoutV2tfp.experimental.nn.ConvolutionVariationalReparameterizationtfp.experimental.nn.ConvolutionVariationalReparameterizationV2tfp.experimental.nn.Layertfp.experimental.nn.Sequentialtfp.experimental.nn.VariationalLayertfp.experimental.nn.initializerstfp.experimental.nn.initializers.glorot_normaltfp.experimental.nn.initializers.glorot_uniformtfp.experimental.nn.initializers.he_normaltfp.experimental.nn.initializers.he_uniformtfp.experimental.nn.lossestfp.experimental.nn.losses.compute_extra_losstfp.experimental.nn.losses.kl_divergence_exacttfp.experimental.nn.losses.kl_divergence_monte_carlotfp.experimental.nn.losses.neglogliktfp.experimental.nn.utiltfp.experimental.nn.util.CallOncetfp.experimental.nn.util.RandomVariabletfp.experimental.nn.util.batchify_optfp.experimental.nn.util.display_imgstfp.experimental.nn.util.expand_dimstfp.experimental.nn.util.flatten_rightmosttfp.experimental.nn.util.halflife_decaytfp.experimental.nn.util.im2rowtfp.experimental.nn.util.im2row_indextfp.experimental.nn.util.make_convolution_fntfp.experimental.nn.util.make_convolution_transpose_fn_with_dilationtfp.experimental.nn.util.make_convolution_transpose_fn_with_subkernelstfp.experimental.nn.util.make_convolution_transpose_fn_with_subkernels_matrixtfp.experimental.nn.util.make_fit_optfp.experimental.nn.util.make_kernel_biastfp.experimental.nn.util.make_kernel_bias_posterior_mvn_diagtfp.experimental.nn.util.make_kernel_bias_prior_spike_and_slabtfp.experimental.nn.util.prepare_conv_argstfp.experimental.nn.util.prepare_tuple_argumenttfp.experimental.nn.util.tfcompiletfp.experimental.nn.util.tracetfp.experimental.nn.util.tune_datasettfp.experimental.nn.util.variables_loadtfp.experimental.nn.util.variables_savetfp.experimental.nn.util.variables_summarytfp.experimental.parallel_filtertfp.experimental.parallel_filter.kalman_filtertfp.experimental.parallel_filter.sample_walktfp.experimental.register_compositetfp.experimental.sequentialtfp.experimental.sequential.EnsembleKalmanFilterStatetfp.experimental.sequential.IteratedFiltertfp.experimental.sequential.ensemble_adjustment_kalman_filter_updatetfp.experimental.sequential.ensemble_kalman_filter_predicttfp.experimental.sequential.ensemble_kalman_filter_updatetfp.experimental.sequential.extended_kalman_filtertfp.experimental.sequential.geometric_cooling_scheduletfp.experimental.sequential.inflate_by_scaled_identity_fntfp.experimental.statstfp.experimental.stats.RunningCentralMomentstfp.experimental.stats.RunningCovariancetfp.experimental.stats.RunningMeantfp.experimental.stats.RunningPotentialScaleReductiontfp.experimental.stats.RunningVariancetfp.experimental.substratestfp.experimental.substrates.jaxtfp.experimental.substrates.jax.bijectorstfp.experimental.substrates.jax.bijectors.AbsoluteValuetfp.experimental.substrates.jax.bijectors.Affinetfp.experimental.substrates.jax.bijectors.AffineLinearOperatortfp.experimental.substrates.jax.bijectors.AffineScalartfp.experimental.substrates.jax.bijectors.Ascendingtfp.experimental.substrates.jax.bijectors.AutoCompositeTensorBijectortfp.experimental.substrates.jax.bijectors.AutoregressiveNetworktfp.experimental.substrates.jax.bijectors.Bijectortfp.experimental.substrates.jax.bijectors.Blockwisetfp.experimental.substrates.jax.bijectors.Chaintfp.experimental.substrates.jax.bijectors.CholeskyOuterProducttfp.experimental.substrates.jax.bijectors.CholeskyToInvCholeskytfp.experimental.substrates.jax.bijectors.Compositiontfp.experimental.substrates.jax.bijectors.CorrelationCholeskytfp.experimental.substrates.jax.bijectors.Cumsumtfp.experimental.substrates.jax.bijectors.Exptfp.experimental.substrates.jax.bijectors.Expm1tfp.experimental.substrates.jax.bijectors.FillScaleTriLtfp.experimental.substrates.jax.bijectors.FillTriangulartfp.experimental.substrates.jax.bijectors.FrechetCDFtfp.experimental.substrates.jax.bijectors.GeneralizedExtremeValueCDFtfp.experimental.substrates.jax.bijectors.GeneralizedParetotfp.experimental.substrates.jax.bijectors.GompertzCDFtfp.experimental.substrates.jax.bijectors.GumbelCDFtfp.experimental.substrates.jax.bijectors.Identitytfp.experimental.substrates.jax.bijectors.Inlinetfp.experimental.substrates.jax.bijectors.Inverttfp.experimental.substrates.jax.bijectors.IteratedSigmoidCenteredtfp.experimental.substrates.jax.bijectors.JointMaptfp.experimental.substrates.jax.bijectors.KumaraswamyCDFtfp.experimental.substrates.jax.bijectors.LambertWTailtfp.experimental.substrates.jax.bijectors.Logtfp.experimental.substrates.jax.bijectors.Log1ptfp.experimental.substrates.jax.bijectors.MaskedAutoregressiveFlowtfp.experimental.substrates.jax.bijectors.MatrixInverseTriLtfp.experimental.substrates.jax.bijectors.MatvecLUtfp.experimental.substrates.jax.bijectors.MoyalCDFtfp.experimental.substrates.jax.bijectors.NormalCDFtfp.experimental.substrates.jax.bijectors.Orderedtfp.experimental.substrates.jax.bijectors.Padtfp.experimental.substrates.jax.bijectors.Permutetfp.experimental.substrates.jax.bijectors.Powertfp.experimental.substrates.jax.bijectors.PowerTransformtfp.experimental.substrates.jax.bijectors.RationalQuadraticSplinetfp.experimental.substrates.jax.bijectors.RayleighCDFtfp.experimental.substrates.jax.bijectors.RealNVPtfp.experimental.substrates.jax.bijectors.Reciprocaltfp.experimental.substrates.jax.bijectors.Reshapetfp.experimental.substrates.jax.bijectors.Restructuretfp.experimental.substrates.jax.bijectors.Scaletfp.experimental.substrates.jax.bijectors.ScaleMatvecDiagtfp.experimental.substrates.jax.bijectors.ScaleMatvecLUtfp.experimental.substrates.jax.bijectors.ScaleMatvecLinearOperatortfp.experimental.substrates.jax.bijectors.ScaleMatvecLinearOperatorBlocktfp.experimental.substrates.jax.bijectors.ScaleMatvecTriLtfp.experimental.substrates.jax.bijectors.Shifttfp.experimental.substrates.jax.bijectors.ShiftedGompertzCDFtfp.experimental.substrates.jax.bijectors.Sigmoidtfp.experimental.substrates.jax.bijectors.Sinhtfp.experimental.substrates.jax.bijectors.SinhArcsinhtfp.experimental.substrates.jax.bijectors.SoftCliptfp.experimental.substrates.jax.bijectors.Softfloortfp.experimental.substrates.jax.bijectors.SoftmaxCenteredtfp.experimental.substrates.jax.bijectors.Softplustfp.experimental.substrates.jax.bijectors.Softsigntfp.experimental.substrates.jax.bijectors.Splittfp.experimental.substrates.jax.bijectors.Squaretfp.experimental.substrates.jax.bijectors.Tanhtfp.experimental.substrates.jax.bijectors.TransformDiagonaltfp.experimental.substrates.jax.bijectors.Transposetfp.experimental.substrates.jax.bijectors.WeibullCDFtfp.experimental.substrates.jax.bijectors.masked_autoregressive_default_templatetfp.experimental.substrates.jax.bijectors.masked_densetfp.experimental.substrates.jax.bijectors.pack_sequence_astfp.experimental.substrates.jax.bijectors.real_nvp_default_templatetfp.experimental.substrates.jax.bijectors.tree_flattentfp.experimental.substrates.jax.distributionstfp.experimental.substrates.jax.distributions.Autoregressivetfp.experimental.substrates.jax.distributions.BatchBroadcasttfp.experimental.substrates.jax.distributions.BatchReshapetfp.experimental.substrates.jax.distributions.Batestfp.experimental.substrates.jax.distributions.Bernoullitfp.experimental.substrates.jax.distributions.Betatfp.experimental.substrates.jax.distributions.BetaBinomialtfp.experimental.substrates.jax.distributions.BetaQuotienttfp.experimental.substrates.jax.distributions.Binomialtfp.experimental.substrates.jax.distributions.Blockwisetfp.experimental.substrates.jax.distributions.Categoricaltfp.experimental.substrates.jax.distributions.Cauchytfp.experimental.substrates.jax.distributions.Chitfp.experimental.substrates.jax.distributions.Chi2tfp.experimental.substrates.jax.distributions.CholeskyLKJtfp.experimental.substrates.jax.distributions.DeterminantalPointProcesstfp.experimental.substrates.jax.distributions.Deterministictfp.experimental.substrates.jax.distributions.Dirichlettfp.experimental.substrates.jax.distributions.DirichletMultinomialtfp.experimental.substrates.jax.distributions.Distributiontfp.experimental.substrates.jax.distributions.DoublesidedMaxwelltfp.experimental.substrates.jax.distributions.Empiricaltfp.experimental.substrates.jax.distributions.ExpGammatfp.experimental.substrates.jax.distributions.ExpInverseGammatfp.experimental.substrates.jax.distributions.ExpRelaxedOneHotCategoricaltfp.experimental.substrates.jax.distributions.Exponentialtfp.experimental.substrates.jax.distributions.ExponentiallyModifiedGaussiantfp.experimental.substrates.jax.distributions.Gammatfp.experimental.substrates.jax.distributions.GammaGammatfp.experimental.substrates.jax.distributions.GaussianProcesstfp.experimental.substrates.jax.distributions.GaussianProcessRegressionModeltfp.experimental.substrates.jax.distributions.GeneralizedExtremeValuetfp.experimental.substrates.jax.distributions.GeneralizedNormaltfp.experimental.substrates.jax.distributions.GeneralizedParetotfp.experimental.substrates.jax.distributions.Geometrictfp.experimental.substrates.jax.distributions.Gumbeltfp.experimental.substrates.jax.distributions.HalfCauchytfp.experimental.substrates.jax.distributions.HalfNormaltfp.experimental.substrates.jax.distributions.HalfStudentTtfp.experimental.substrates.jax.distributions.HiddenMarkovModeltfp.experimental.substrates.jax.distributions.Horseshoetfp.experimental.substrates.jax.distributions.Independenttfp.experimental.substrates.jax.distributions.InverseGammatfp.experimental.substrates.jax.distributions.InverseGaussiantfp.experimental.substrates.jax.distributions.JohnsonSUtfp.experimental.substrates.jax.distributions.JointDistributiontfp.experimental.substrates.jax.distributions.JointDistribution.Roottfp.experimental.substrates.jax.distributions.JointDistributionCoroutinetfp.experimental.substrates.jax.distributions.JointDistributionCoroutine.Roottfp.experimental.substrates.jax.distributions.JointDistributionCoroutineAutoBatchedtfp.experimental.substrates.jax.distributions.JointDistributionCoroutineAutoBatched.Roottfp.experimental.substrates.jax.distributions.JointDistributionNamedtfp.experimental.substrates.jax.distributions.JointDistributionNamed.Roottfp.experimental.substrates.jax.distributions.JointDistributionNamedAutoBatchedtfp.experimental.substrates.jax.distributions.JointDistributionNamedAutoBatched.Roottfp.experimental.substrates.jax.distributions.JointDistributionSequentialtfp.experimental.substrates.jax.distributions.JointDistributionSequential.Roottfp.experimental.substrates.jax.distributions.JointDistributionSequentialAutoBatchedtfp.experimental.substrates.jax.distributions.JointDistributionSequentialAutoBatched.Roottfp.experimental.substrates.jax.distributions.Kumaraswamytfp.experimental.substrates.jax.distributions.LKJtfp.experimental.substrates.jax.distributions.LambertWDistributiontfp.experimental.substrates.jax.distributions.LambertWNormaltfp.experimental.substrates.jax.distributions.Laplacetfp.experimental.substrates.jax.distributions.LinearGaussianStateSpaceModeltfp.experimental.substrates.jax.distributions.LogLogistictfp.experimental.substrates.jax.distributions.LogNormaltfp.experimental.substrates.jax.distributions.Logistictfp.experimental.substrates.jax.distributions.LogitNormaltfp.experimental.substrates.jax.distributions.Maskedtfp.experimental.substrates.jax.distributions.MatrixNormalLinearOperatortfp.experimental.substrates.jax.distributions.MatrixTLinearOperatortfp.experimental.substrates.jax.distributions.MixtureSameFamilytfp.experimental.substrates.jax.distributions.Moyaltfp.experimental.substrates.jax.distributions.Multinomialtfp.experimental.substrates.jax.distributions.MultivariateNormalDiagtfp.experimental.substrates.jax.distributions.MultivariateNormalDiagPlusLowRanktfp.experimental.substrates.jax.distributions.MultivariateNormalFullCovariancetfp.experimental.substrates.jax.distributions.MultivariateNormalLinearOperatortfp.experimental.substrates.jax.distributions.MultivariateNormalTriLtfp.experimental.substrates.jax.distributions.MultivariateStudentTLinearOperatortfp.experimental.substrates.jax.distributions.NegativeBinomialtfp.experimental.substrates.jax.distributions.Normaltfp.experimental.substrates.jax.distributions.NormalInverseGaussiantfp.experimental.substrates.jax.distributions.OneHotCategoricaltfp.experimental.substrates.jax.distributions.OrderedLogistictfp.experimental.substrates.jax.distributions.PERTtfp.experimental.substrates.jax.distributions.Paretotfp.experimental.substrates.jax.distributions.PlackettLucetfp.experimental.substrates.jax.distributions.Poissontfp.experimental.substrates.jax.distributions.PoissonLogNormalQuadratureCompoundtfp.experimental.substrates.jax.distributions.PowerSphericaltfp.experimental.substrates.jax.distributions.ProbitBernoullitfp.experimental.substrates.jax.distributions.QuantizedDistributiontfp.experimental.substrates.jax.distributions.RegisterKLtfp.experimental.substrates.jax.distributions.RelaxedBernoullitfp.experimental.substrates.jax.distributions.RelaxedOneHotCategoricaltfp.experimental.substrates.jax.distributions.ReparameterizationTypetfp.experimental.substrates.jax.distributions.Sampletfp.experimental.substrates.jax.distributions.SigmoidBetatfp.experimental.substrates.jax.distributions.SinhArcsinhtfp.experimental.substrates.jax.distributions.Skellamtfp.experimental.substrates.jax.distributions.SphericalUniformtfp.experimental.substrates.jax.distributions.StoppingRatioLogistictfp.experimental.substrates.jax.distributions.StudentTtfp.experimental.substrates.jax.distributions.StudentTProcesstfp.experimental.substrates.jax.distributions.TransformedDistributiontfp.experimental.substrates.jax.distributions.Triangulartfp.experimental.substrates.jax.distributions.TruncatedCauchytfp.experimental.substrates.jax.distributions.TruncatedNormaltfp.experimental.substrates.jax.distributions.Uniformtfp.experimental.substrates.jax.distributions.VariationalGaussianProcesstfp.experimental.substrates.jax.distributions.VectorDeterministictfp.experimental.substrates.jax.distributions.VectorExponentialDiagtfp.experimental.substrates.jax.distributions.VonMisestfp.experimental.substrates.jax.distributions.VonMisesFishertfp.experimental.substrates.jax.distributions.Weibulltfp.experimental.substrates.jax.distributions.WishartLinearOperatortfp.experimental.substrates.jax.distributions.WishartTriLtfp.experimental.substrates.jax.distributions.Zipftfp.experimental.substrates.jax.distributions.independent_joint_distribution_from_structuretfp.experimental.substrates.jax.distributions.kl_divergencetfp.experimental.substrates.jax.distributions.mvn_conjugate_linear_updatetfp.experimental.substrates.jax.distributions.normal_conjugates_known_scale_posteriortfp.experimental.substrates.jax.distributions.normal_conjugates_known_scale_predictivetfp.experimental.substrates.jax.distributions.quadrature_scheme_lognormal_gauss_hermitetfp.experimental.substrates.jax.distributions.quadrature_scheme_lognormal_quantilestfp.experimental.substrates.jax.mathtfp.experimental.substrates.jax.math.atan_differencetfp.experimental.substrates.jax.math.batch_interp_regular_1d_gridtfp.experimental.substrates.jax.math.batch_interp_regular_nd_gridtfp.experimental.substrates.jax.math.bessel_iv_ratiotfp.experimental.substrates.jax.math.bessel_ivetfp.experimental.substrates.jax.math.bessel_kvetfp.experimental.substrates.jax.math.bracket_roottfp.experimental.substrates.jax.math.cholesky_concattfp.experimental.substrates.jax.math.cholesky_updatetfp.experimental.substrates.jax.math.clip_by_value_preserve_gradienttfp.experimental.substrates.jax.math.custom_gradienttfp.experimental.substrates.jax.math.dawsntfp.experimental.substrates.jax.math.diag_jacobiantfp.experimental.substrates.jax.math.erfcinvtfp.experimental.substrates.jax.math.erfcxtfp.experimental.substrates.jax.math.fill_triangulartfp.experimental.substrates.jax.math.fill_triangular_inversetfp.experimental.substrates.jax.math.find_root_chandrupatlatfp.experimental.substrates.jax.math.find_root_secanttfp.experimental.substrates.jax.math.gram_schmidttfp.experimental.substrates.jax.math.hypergeometrictfp.experimental.substrates.jax.math.hypergeometric.hyp2f1_small_argumenttfp.experimental.substrates.jax.math.igammacinvtfp.experimental.substrates.jax.math.igammainvtfp.experimental.substrates.jax.math.interp_regular_1d_gridtfp.experimental.substrates.jax.math.lambertwtfp.experimental.substrates.jax.math.lambertw_winitzki_approxtfp.experimental.substrates.jax.math.lbetatfp.experimental.substrates.jax.math.log1mexptfp.experimental.substrates.jax.math.log1psquaretfp.experimental.substrates.jax.math.log_add_exptfp.experimental.substrates.jax.math.log_bessel_ivetfp.experimental.substrates.jax.math.log_bessel_kvetfp.experimental.substrates.jax.math.log_combinationstfp.experimental.substrates.jax.math.log_coshtfp.experimental.substrates.jax.math.log_cumsum_exptfp.experimental.substrates.jax.math.log_gamma_correctiontfp.experimental.substrates.jax.math.log_gamma_differencetfp.experimental.substrates.jax.math.log_sub_exptfp.experimental.substrates.jax.math.logerfctfp.experimental.substrates.jax.math.logerfcxtfp.experimental.substrates.jax.math.lu_matrix_inversetfp.experimental.substrates.jax.math.lu_reconstructtfp.experimental.substrates.jax.math.lu_solvetfp.experimental.substrates.jax.math.owens_ttfp.experimental.substrates.jax.math.pivoted_choleskytfp.experimental.substrates.jax.math.psd_kernelstfp.experimental.substrates.jax.math.psd_kernels.ExpSinSquaredtfp.experimental.substrates.jax.math.psd_kernels.ExponentiatedQuadratictfp.experimental.substrates.jax.math.psd_kernels.FeatureScaledtfp.experimental.substrates.jax.math.psd_kernels.FeatureTransformedtfp.experimental.substrates.jax.math.psd_kernels.GeneralizedMaterntfp.experimental.substrates.jax.math.psd_kernels.KumaraswamyTransformedtfp.experimental.substrates.jax.math.psd_kernels.Lineartfp.experimental.substrates.jax.math.psd_kernels.MaternFiveHalvestfp.experimental.substrates.jax.math.psd_kernels.MaternOneHalftfp.experimental.substrates.jax.math.psd_kernels.MaternThreeHalvestfp.experimental.substrates.jax.math.psd_kernels.Parabolictfp.experimental.substrates.jax.math.psd_kernels.Polynomialtfp.experimental.substrates.jax.math.psd_kernels.PositiveSemidefiniteKerneltfp.experimental.substrates.jax.math.psd_kernels.RationalQuadratictfp.experimental.substrates.jax.math.psd_kernels.SchurComplementtfp.experimental.substrates.jax.math.random_rademachertfp.experimental.substrates.jax.math.random_rayleightfp.experimental.substrates.jax.math.reduce_kahan_sumtfp.experimental.substrates.jax.math.reduce_log_harmonic_mean_exptfp.experimental.substrates.jax.math.reduce_logmeanexptfp.experimental.substrates.jax.math.reduce_weighted_logsumexptfp.experimental.substrates.jax.math.round_exponential_bump_functiontfp.experimental.substrates.jax.math.scan_associativetfp.experimental.substrates.jax.math.secant_roottfp.experimental.substrates.jax.math.smoothersteptfp.experimental.substrates.jax.math.soft_sorting_matrixtfp.experimental.substrates.jax.math.soft_thresholdtfp.experimental.substrates.jax.math.softplus_inversetfp.experimental.substrates.jax.math.sparse_or_dense_matmultfp.experimental.substrates.jax.math.sparse_or_dense_matvecmultfp.experimental.substrates.jax.math.sqrt1pm1tfp.experimental.substrates.jax.math.trapztfp.experimental.substrates.jax.math.value_and_gradienttfp.experimental.substrates.jax.mcmctfp.experimental.substrates.jax.mcmc.CheckpointableStatesAndTracetfp.experimental.substrates.jax.mcmc.DualAveragingStepSizeAdaptationtfp.experimental.substrates.jax.mcmc.HamiltonianMonteCarlotfp.experimental.substrates.jax.mcmc.MetropolisAdjustedLangevinAlgorithmtfp.experimental.substrates.jax.mcmc.MetropolisHastingstfp.experimental.substrates.jax.mcmc.NoUTurnSamplertfp.experimental.substrates.jax.mcmc.RandomWalkMetropolistfp.experimental.substrates.jax.mcmc.ReplicaExchangeMCtfp.experimental.substrates.jax.mcmc.SimpleStepSizeAdaptationtfp.experimental.substrates.jax.mcmc.SliceSamplertfp.experimental.substrates.jax.mcmc.StatesAndTracetfp.experimental.substrates.jax.mcmc.TransformedTransitionKerneltfp.experimental.substrates.jax.mcmc.TransitionKerneltfp.experimental.substrates.jax.mcmc.UncalibratedHamiltonianMonteCarlotfp.experimental.substrates.jax.mcmc.UncalibratedLangevintfp.experimental.substrates.jax.mcmc.UncalibratedRandomWalktfp.experimental.substrates.jax.mcmc.default_swap_proposal_fntfp.experimental.substrates.jax.mcmc.effective_sample_sizetfp.experimental.substrates.jax.mcmc.even_odd_swap_proposal_fntfp.experimental.substrates.jax.mcmc.make_simple_step_size_update_policytfp.experimental.substrates.jax.mcmc.potential_scale_reductiontfp.experimental.substrates.jax.mcmc.random_walk_normal_fntfp.experimental.substrates.jax.mcmc.random_walk_uniform_fntfp.experimental.substrates.jax.mcmc.sample_annealed_importance_chaintfp.experimental.substrates.jax.mcmc.sample_chaintfp.experimental.substrates.jax.mcmc.sample_halton_sequencetfp.experimental.substrates.jax.optimizertfp.experimental.substrates.jax.optimizer.bfgs_minimizetfp.experimental.substrates.jax.optimizer.converged_alltfp.experimental.substrates.jax.optimizer.converged_anytfp.experimental.substrates.jax.optimizer.lbfgs_minimizetfp.experimental.substrates.jax.optimizer.linesearchtfp.experimental.substrates.jax.optimizer.linesearch.hager_zhangtfp.experimental.substrates.jax.optimizer.nelder_mead_minimizetfp.experimental.substrates.jax.optimizer.nelder_mead_one_steptfp.experimental.substrates.jax.randomtfp.experimental.substrates.jax.random.rademachertfp.experimental.substrates.jax.random.rayleightfp.experimental.substrates.jax.random.sanitize_seedtfp.experimental.substrates.jax.random.spherical_uniformtfp.experimental.substrates.jax.random.split_seedtfp.experimental.substrates.jax.statstfp.experimental.substrates.jax.stats.assign_log_moving_mean_exptfp.experimental.substrates.jax.stats.assign_moving_mean_variancetfp.experimental.substrates.jax.stats.auto_correlationtfp.experimental.substrates.jax.stats.brier_decompositiontfp.experimental.substrates.jax.stats.brier_scoretfp.experimental.substrates.jax.stats.cholesky_covariancetfp.experimental.substrates.jax.stats.correlationtfp.experimental.substrates.jax.stats.count_integerstfp.experimental.substrates.jax.stats.covariancetfp.experimental.substrates.jax.stats.cumulative_variancetfp.experimental.substrates.jax.stats.expected_calibration_errortfp.experimental.substrates.jax.stats.expected_calibration_error_quantilestfp.experimental.substrates.jax.stats.find_binstfp.experimental.substrates.jax.stats.histogramtfp.experimental.substrates.jax.stats.iterative_mergesorttfp.experimental.substrates.jax.stats.kendalls_tautfp.experimental.substrates.jax.stats.lexicographical_indirect_sorttfp.experimental.substrates.jax.stats.log_average_probstfp.experimental.substrates.jax.stats.log_loomean_exptfp.experimental.substrates.jax.stats.log_loosum_exptfp.experimental.substrates.jax.stats.log_soomean_exptfp.experimental.substrates.jax.stats.log_soosum_exptfp.experimental.substrates.jax.stats.moving_mean_variance_zero_debiasedtfp.experimental.substrates.jax.stats.percentiletfp.experimental.substrates.jax.stats.quantilestfp.experimental.substrates.jax.stats.stddevtfp.experimental.substrates.jax.stats.variancetfp.experimental.substrates.jax.stats.windowed_meantfp.experimental.substrates.jax.stats.windowed_variancetfp.experimental.substrates.jax.utiltfp.experimental.substrates.jax.util.DeferredTensortfp.experimental.substrates.jax.util.ParameterPropertiestfp.experimental.substrates.jax.util.SeedStreamtfp.experimental.substrates.jax.util.TransformedVariabletfp.experimental.substrates.numpytfp.experimental.substrates.numpy.bijectorstfp.experimental.substrates.numpy.bijectors.AbsoluteValuetfp.experimental.substrates.numpy.bijectors.Affinetfp.experimental.substrates.numpy.bijectors.AffineLinearOperatortfp.experimental.substrates.numpy.bijectors.AffineScalartfp.experimental.substrates.numpy.bijectors.Ascendingtfp.experimental.substrates.numpy.bijectors.AutoCompositeTensorBijectortfp.experimental.substrates.numpy.bijectors.AutoregressiveNetworktfp.experimental.substrates.numpy.bijectors.Bijectortfp.experimental.substrates.numpy.bijectors.Blockwisetfp.experimental.substrates.numpy.bijectors.Chaintfp.experimental.substrates.numpy.bijectors.CholeskyOuterProducttfp.experimental.substrates.numpy.bijectors.CholeskyToInvCholeskytfp.experimental.substrates.numpy.bijectors.Compositiontfp.experimental.substrates.numpy.bijectors.CorrelationCholeskytfp.experimental.substrates.numpy.bijectors.Cumsumtfp.experimental.substrates.numpy.bijectors.Exptfp.experimental.substrates.numpy.bijectors.Expm1tfp.experimental.substrates.numpy.bijectors.FillScaleTriLtfp.experimental.substrates.numpy.bijectors.FillTriangulartfp.experimental.substrates.numpy.bijectors.FrechetCDFtfp.experimental.substrates.numpy.bijectors.GeneralizedExtremeValueCDFtfp.experimental.substrates.numpy.bijectors.GeneralizedParetotfp.experimental.substrates.numpy.bijectors.GompertzCDFtfp.experimental.substrates.numpy.bijectors.GumbelCDFtfp.experimental.substrates.numpy.bijectors.Identitytfp.experimental.substrates.numpy.bijectors.Inlinetfp.experimental.substrates.numpy.bijectors.Inverttfp.experimental.substrates.numpy.bijectors.IteratedSigmoidCenteredtfp.experimental.substrates.numpy.bijectors.JointMaptfp.experimental.substrates.numpy.bijectors.KumaraswamyCDFtfp.experimental.substrates.numpy.bijectors.LambertWTailtfp.experimental.substrates.numpy.bijectors.Logtfp.experimental.substrates.numpy.bijectors.Log1ptfp.experimental.substrates.numpy.bijectors.MaskedAutoregressiveFlowtfp.experimental.substrates.numpy.bijectors.MatrixInverseTriLtfp.experimental.substrates.numpy.bijectors.MatvecLUtfp.experimental.substrates.numpy.bijectors.MoyalCDFtfp.experimental.substrates.numpy.bijectors.NormalCDFtfp.experimental.substrates.numpy.bijectors.Orderedtfp.experimental.substrates.numpy.bijectors.Padtfp.experimental.substrates.numpy.bijectors.Permutetfp.experimental.substrates.numpy.bijectors.Powertfp.experimental.substrates.numpy.bijectors.PowerTransformtfp.experimental.substrates.numpy.bijectors.RationalQuadraticSplinetfp.experimental.substrates.numpy.bijectors.RayleighCDFtfp.experimental.substrates.numpy.bijectors.RealNVPtfp.experimental.substrates.numpy.bijectors.Reciprocaltfp.experimental.substrates.numpy.bijectors.Reshapetfp.experimental.substrates.numpy.bijectors.Restructuretfp.experimental.substrates.numpy.bijectors.Scaletfp.experimental.substrates.numpy.bijectors.ScaleMatvecDiagtfp.experimental.substrates.numpy.bijectors.ScaleMatvecLUtfp.experimental.substrates.numpy.bijectors.ScaleMatvecLinearOperatortfp.experimental.substrates.numpy.bijectors.ScaleMatvecLinearOperatorBlocktfp.experimental.substrates.numpy.bijectors.ScaleMatvecTriLtfp.experimental.substrates.numpy.bijectors.Shifttfp.experimental.substrates.numpy.bijectors.ShiftedGompertzCDFtfp.experimental.substrates.numpy.bijectors.Sigmoidtfp.experimental.substrates.numpy.bijectors.Sinhtfp.experimental.substrates.numpy.bijectors.SinhArcsinhtfp.experimental.substrates.numpy.bijectors.SoftCliptfp.experimental.substrates.numpy.bijectors.Softfloortfp.experimental.substrates.numpy.bijectors.SoftmaxCenteredtfp.experimental.substrates.numpy.bijectors.Softplustfp.experimental.substrates.numpy.bijectors.Softsigntfp.experimental.substrates.numpy.bijectors.Splittfp.experimental.substrates.numpy.bijectors.Squaretfp.experimental.substrates.numpy.bijectors.Tanhtfp.experimental.substrates.numpy.bijectors.TransformDiagonaltfp.experimental.substrates.numpy.bijectors.Transposetfp.experimental.substrates.numpy.bijectors.WeibullCDFtfp.experimental.substrates.numpy.bijectors.masked_autoregressive_default_templatetfp.experimental.substrates.numpy.bijectors.masked_densetfp.experimental.substrates.numpy.bijectors.pack_sequence_astfp.experimental.substrates.numpy.bijectors.real_nvp_default_templatetfp.experimental.substrates.numpy.bijectors.tree_flattentfp.experimental.substrates.numpy.distributionstfp.experimental.substrates.numpy.distributions.Autoregressivetfp.experimental.substrates.numpy.distributions.BatchBroadcasttfp.experimental.substrates.numpy.distributions.BatchReshapetfp.experimental.substrates.numpy.distributions.Batestfp.experimental.substrates.numpy.distributions.Bernoullitfp.experimental.substrates.numpy.distributions.Betatfp.experimental.substrates.numpy.distributions.BetaBinomialtfp.experimental.substrates.numpy.distributions.BetaQuotienttfp.experimental.substrates.numpy.distributions.Binomialtfp.experimental.substrates.numpy.distributions.Blockwisetfp.experimental.substrates.numpy.distributions.Categoricaltfp.experimental.substrates.numpy.distributions.Cauchytfp.experimental.substrates.numpy.distributions.Chitfp.experimental.substrates.numpy.distributions.Chi2tfp.experimental.substrates.numpy.distributions.CholeskyLKJtfp.experimental.substrates.numpy.distributions.DeterminantalPointProcesstfp.experimental.substrates.numpy.distributions.Deterministictfp.experimental.substrates.numpy.distributions.Dirichlettfp.experimental.substrates.numpy.distributions.DirichletMultinomialtfp.experimental.substrates.numpy.distributions.Distributiontfp.experimental.substrates.numpy.distributions.DoublesidedMaxwelltfp.experimental.substrates.numpy.distributions.Empiricaltfp.experimental.substrates.numpy.distributions.ExpGammatfp.experimental.substrates.numpy.distributions.ExpInverseGammatfp.experimental.substrates.numpy.distributions.ExpRelaxedOneHotCategoricaltfp.experimental.substrates.numpy.distributions.Exponentialtfp.experimental.substrates.numpy.distributions.ExponentiallyModifiedGaussiantfp.experimental.substrates.numpy.distributions.Gammatfp.experimental.substrates.numpy.distributions.GammaGammatfp.experimental.substrates.numpy.distributions.GaussianProcesstfp.experimental.substrates.numpy.distributions.GaussianProcessRegressionModeltfp.experimental.substrates.numpy.distributions.GeneralizedExtremeValuetfp.experimental.substrates.numpy.distributions.GeneralizedNormaltfp.experimental.substrates.numpy.distributions.GeneralizedParetotfp.experimental.substrates.numpy.distributions.Geometrictfp.experimental.substrates.numpy.distributions.Gumbeltfp.experimental.substrates.numpy.distributions.HalfCauchytfp.experimental.substrates.numpy.distributions.HalfNormaltfp.experimental.substrates.numpy.distributions.HalfStudentTtfp.experimental.substrates.numpy.distributions.HiddenMarkovModeltfp.experimental.substrates.numpy.distributions.Horseshoetfp.experimental.substrates.numpy.distributions.Independenttfp.experimental.substrates.numpy.distributions.InverseGammatfp.experimental.substrates.numpy.distributions.InverseGaussiantfp.experimental.substrates.numpy.distributions.JohnsonSUtfp.experimental.substrates.numpy.distributions.JointDistributiontfp.experimental.substrates.numpy.distributions.JointDistribution.Roottfp.experimental.substrates.numpy.distributions.JointDistributionCoroutinetfp.experimental.substrates.numpy.distributions.JointDistributionCoroutine.Roottfp.experimental.substrates.numpy.distributions.JointDistributionCoroutineAutoBatchedtfp.experimental.substrates.numpy.distributions.JointDistributionCoroutineAutoBatched.Roottfp.experimental.substrates.numpy.distributions.JointDistributionNamedtfp.experimental.substrates.numpy.distributions.JointDistributionNamed.Roottfp.experimental.substrates.numpy.distributions.JointDistributionNamedAutoBatchedtfp.experimental.substrates.numpy.distributions.JointDistributionNamedAutoBatched.Roottfp.experimental.substrates.numpy.distributions.JointDistributionSequentialtfp.experimental.substrates.numpy.distributions.JointDistributionSequential.Roottfp.experimental.substrates.numpy.distributions.JointDistributionSequentialAutoBatchedtfp.experimental.substrates.numpy.distributions.JointDistributionSequentialAutoBatched.Roottfp.experimental.substrates.numpy.distributions.Kumaraswamytfp.experimental.substrates.numpy.distributions.LKJtfp.experimental.substrates.numpy.distributions.LambertWDistributiontfp.experimental.substrates.numpy.distributions.LambertWNormaltfp.experimental.substrates.numpy.distributions.Laplacetfp.experimental.substrates.numpy.distributions.LinearGaussianStateSpaceModeltfp.experimental.substrates.numpy.distributions.LogLogistictfp.experimental.substrates.numpy.distributions.LogNormaltfp.experimental.substrates.numpy.distributions.Logistictfp.experimental.substrates.numpy.distributions.LogitNormaltfp.experimental.substrates.numpy.distributions.Maskedtfp.experimental.substrates.numpy.distributions.MatrixNormalLinearOperatortfp.experimental.substrates.numpy.distributions.MatrixTLinearOperatortfp.experimental.substrates.numpy.distributions.MixtureSameFamilytfp.experimental.substrates.numpy.distributions.Moyaltfp.experimental.substrates.numpy.distributions.Multinomialtfp.experimental.substrates.numpy.distributions.MultivariateNormalDiagtfp.experimental.substrates.numpy.distributions.MultivariateNormalDiagPlusLowRanktfp.experimental.substrates.numpy.distributions.MultivariateNormalFullCovariancetfp.experimental.substrates.numpy.distributions.MultivariateNormalLinearOperatortfp.experimental.substrates.numpy.distributions.MultivariateNormalTriLtfp.experimental.substrates.numpy.distributions.MultivariateStudentTLinearOperatortfp.experimental.substrates.numpy.distributions.NegativeBinomialtfp.experimental.substrates.numpy.distributions.Normaltfp.experimental.substrates.numpy.distributions.NormalInverseGaussiantfp.experimental.substrates.numpy.distributions.OneHotCategoricaltfp.experimental.substrates.numpy.distributions.OrderedLogistictfp.experimental.substrates.numpy.distributions.PERTtfp.experimental.substrates.numpy.distributions.Paretotfp.experimental.substrates.numpy.distributions.PlackettLucetfp.experimental.substrates.numpy.distributions.Poissontfp.experimental.substrates.numpy.distributions.PoissonLogNormalQuadratureCompoundtfp.experimental.substrates.numpy.distributions.PowerSphericaltfp.experimental.substrates.numpy.distributions.ProbitBernoullitfp.experimental.substrates.numpy.distributions.QuantizedDistributiontfp.experimental.substrates.numpy.distributions.RegisterKLtfp.experimental.substrates.numpy.distributions.RelaxedBernoullitfp.experimental.substrates.numpy.distributions.RelaxedOneHotCategoricaltfp.experimental.substrates.numpy.distributions.ReparameterizationTypetfp.experimental.substrates.numpy.distributions.Sampletfp.experimental.substrates.numpy.distributions.SigmoidBetatfp.experimental.substrates.numpy.distributions.SinhArcsinhtfp.experimental.substrates.numpy.distributions.Skellamtfp.experimental.substrates.numpy.distributions.SphericalUniformtfp.experimental.substrates.numpy.distributions.StoppingRatioLogistictfp.experimental.substrates.numpy.distributions.StudentTtfp.experimental.substrates.numpy.distributions.StudentTProcesstfp.experimental.substrates.numpy.distributions.TransformedDistributiontfp.experimental.substrates.numpy.distributions.Triangulartfp.experimental.substrates.numpy.distributions.TruncatedCauchytfp.experimental.substrates.numpy.distributions.TruncatedNormaltfp.experimental.substrates.numpy.distributions.Uniformtfp.experimental.substrates.numpy.distributions.VariationalGaussianProcesstfp.experimental.substrates.numpy.distributions.VectorDeterministictfp.experimental.substrates.numpy.distributions.VectorExponentialDiagtfp.experimental.substrates.numpy.distributions.VonMisestfp.experimental.substrates.numpy.distributions.VonMisesFishertfp.experimental.substrates.numpy.distributions.Weibulltfp.experimental.substrates.numpy.distributions.WishartLinearOperatortfp.experimental.substrates.numpy.distributions.WishartTriLtfp.experimental.substrates.numpy.distributions.Zipftfp.experimental.substrates.numpy.distributions.independent_joint_distribution_from_structuretfp.experimental.substrates.numpy.distributions.kl_divergencetfp.experimental.substrates.numpy.distributions.mvn_conjugate_linear_updatetfp.experimental.substrates.numpy.distributions.normal_conjugates_known_scale_posteriortfp.experimental.substrates.numpy.distributions.normal_conjugates_known_scale_predictivetfp.experimental.substrates.numpy.distributions.quadrature_scheme_lognormal_gauss_hermitetfp.experimental.substrates.numpy.distributions.quadrature_scheme_lognormal_quantilestfp.experimental.substrates.numpy.mathtfp.experimental.substrates.numpy.math.atan_differencetfp.experimental.substrates.numpy.math.batch_interp_regular_1d_gridtfp.experimental.substrates.numpy.math.batch_interp_regular_nd_gridtfp.experimental.substrates.numpy.math.bessel_iv_ratiotfp.experimental.substrates.numpy.math.bessel_ivetfp.experimental.substrates.numpy.math.bessel_kvetfp.experimental.substrates.numpy.math.bracket_roottfp.experimental.substrates.numpy.math.cholesky_concattfp.experimental.substrates.numpy.math.cholesky_updatetfp.experimental.substrates.numpy.math.clip_by_value_preserve_gradienttfp.experimental.substrates.numpy.math.custom_gradienttfp.experimental.substrates.numpy.math.dawsntfp.experimental.substrates.numpy.math.diag_jacobiantfp.experimental.substrates.numpy.math.erfcinvtfp.experimental.substrates.numpy.math.erfcxtfp.experimental.substrates.numpy.math.fill_triangulartfp.experimental.substrates.numpy.math.fill_triangular_inversetfp.experimental.substrates.numpy.math.find_root_chandrupatlatfp.experimental.substrates.numpy.math.find_root_secanttfp.experimental.substrates.numpy.math.gram_schmidttfp.experimental.substrates.numpy.math.hypergeometrictfp.experimental.substrates.numpy.math.hypergeometric.hyp2f1_small_argumenttfp.experimental.substrates.numpy.math.igammacinvtfp.experimental.substrates.numpy.math.igammainvtfp.experimental.substrates.numpy.math.interp_regular_1d_gridtfp.experimental.substrates.numpy.math.lambertwtfp.experimental.substrates.numpy.math.lambertw_winitzki_approxtfp.experimental.substrates.numpy.math.lbetatfp.experimental.substrates.numpy.math.log1mexptfp.experimental.substrates.numpy.math.log1psquaretfp.experimental.substrates.numpy.math.log_add_exptfp.experimental.substrates.numpy.math.log_bessel_ivetfp.experimental.substrates.numpy.math.log_bessel_kvetfp.experimental.substrates.numpy.math.log_combinationstfp.experimental.substrates.numpy.math.log_coshtfp.experimental.substrates.numpy.math.log_cumsum_exptfp.experimental.substrates.numpy.math.log_gamma_correctiontfp.experimental.substrates.numpy.math.log_gamma_differencetfp.experimental.substrates.numpy.math.log_sub_exptfp.experimental.substrates.numpy.math.logerfctfp.experimental.substrates.numpy.math.logerfcxtfp.experimental.substrates.numpy.math.lu_matrix_inversetfp.experimental.substrates.numpy.math.lu_reconstructtfp.experimental.substrates.numpy.math.lu_solvetfp.experimental.substrates.numpy.math.owens_ttfp.experimental.substrates.numpy.math.pivoted_choleskytfp.experimental.substrates.numpy.math.psd_kernelstfp.experimental.substrates.numpy.math.psd_kernels.ExpSinSquaredtfp.experimental.substrates.numpy.math.psd_kernels.ExponentiatedQuadratictfp.experimental.substrates.numpy.math.psd_kernels.FeatureScaledtfp.experimental.substrates.numpy.math.psd_kernels.FeatureTransformedtfp.experimental.substrates.numpy.math.psd_kernels.GeneralizedMaterntfp.experimental.substrates.numpy.math.psd_kernels.KumaraswamyTransformedtfp.experimental.substrates.numpy.math.psd_kernels.Lineartfp.experimental.substrates.numpy.math.psd_kernels.MaternFiveHalvestfp.experimental.substrates.numpy.math.psd_kernels.MaternOneHalftfp.experimental.substrates.numpy.math.psd_kernels.MaternThreeHalvestfp.experimental.substrates.numpy.math.psd_kernels.Parabolictfp.experimental.substrates.numpy.math.psd_kernels.Polynomialtfp.experimental.substrates.numpy.math.psd_kernels.PositiveSemidefiniteKerneltfp.experimental.substrates.numpy.math.psd_kernels.RationalQuadratictfp.experimental.substrates.numpy.math.psd_kernels.SchurComplementtfp.experimental.substrates.numpy.math.random_rademachertfp.experimental.substrates.numpy.math.random_rayleightfp.experimental.substrates.numpy.math.reduce_kahan_sumtfp.experimental.substrates.numpy.math.reduce_log_harmonic_mean_exptfp.experimental.substrates.numpy.math.reduce_logmeanexptfp.experimental.substrates.numpy.math.reduce_weighted_logsumexptfp.experimental.substrates.numpy.math.round_exponential_bump_functiontfp.experimental.substrates.numpy.math.scan_associativetfp.experimental.substrates.numpy.math.secant_roottfp.experimental.substrates.numpy.math.smoothersteptfp.experimental.substrates.numpy.math.soft_sorting_matrixtfp.experimental.substrates.numpy.math.soft_thresholdtfp.experimental.substrates.numpy.math.softplus_inversetfp.experimental.substrates.numpy.math.sparse_or_dense_matmultfp.experimental.substrates.numpy.math.sparse_or_dense_matvecmultfp.experimental.substrates.numpy.math.sqrt1pm1tfp.experimental.substrates.numpy.math.trapztfp.experimental.substrates.numpy.math.value_and_gradienttfp.experimental.substrates.numpy.mcmctfp.experimental.substrates.numpy.mcmc.CheckpointableStatesAndTracetfp.experimental.substrates.numpy.mcmc.DualAveragingStepSizeAdaptationtfp.experimental.substrates.numpy.mcmc.MetropolisHastingstfp.experimental.substrates.numpy.mcmc.RandomWalkMetropolistfp.experimental.substrates.numpy.mcmc.ReplicaExchangeMCtfp.experimental.substrates.numpy.mcmc.SimpleStepSizeAdaptationtfp.experimental.substrates.numpy.mcmc.SliceSamplertfp.experimental.substrates.numpy.mcmc.StatesAndTracetfp.experimental.substrates.numpy.mcmc.TransformedTransitionKerneltfp.experimental.substrates.numpy.mcmc.TransitionKerneltfp.experimental.substrates.numpy.mcmc.UncalibratedRandomWalktfp.experimental.substrates.numpy.mcmc.default_swap_proposal_fntfp.experimental.substrates.numpy.mcmc.effective_sample_sizetfp.experimental.substrates.numpy.mcmc.even_odd_swap_proposal_fntfp.experimental.substrates.numpy.mcmc.potential_scale_reductiontfp.experimental.substrates.numpy.mcmc.random_walk_normal_fntfp.experimental.substrates.numpy.mcmc.random_walk_uniform_fntfp.experimental.substrates.numpy.mcmc.sample_annealed_importance_chaintfp.experimental.substrates.numpy.mcmc.sample_chaintfp.experimental.substrates.numpy.mcmc.sample_halton_sequencetfp.experimental.substrates.numpy.optimizertfp.experimental.substrates.numpy.optimizer.bfgs_minimizetfp.experimental.substrates.numpy.optimizer.converged_alltfp.experimental.substrates.numpy.optimizer.converged_anytfp.experimental.substrates.numpy.optimizer.lbfgs_minimizetfp.experimental.substrates.numpy.optimizer.linesearchtfp.experimental.substrates.numpy.optimizer.linesearch.hager_zhangtfp.experimental.substrates.numpy.optimizer.nelder_mead_minimizetfp.experimental.substrates.numpy.optimizer.nelder_mead_one_steptfp.experimental.substrates.numpy.randomtfp.experimental.substrates.numpy.random.rademachertfp.experimental.substrates.numpy.random.rayleightfp.experimental.substrates.numpy.random.sanitize_seedtfp.experimental.substrates.numpy.random.spherical_uniformtfp.experimental.substrates.numpy.random.split_seedtfp.experimental.substrates.numpy.statstfp.experimental.substrates.numpy.stats.assign_log_moving_mean_exptfp.experimental.substrates.numpy.stats.assign_moving_mean_variancetfp.experimental.substrates.numpy.stats.auto_correlationtfp.experimental.substrates.numpy.stats.brier_decompositiontfp.experimental.substrates.numpy.stats.brier_scoretfp.experimental.substrates.numpy.stats.cholesky_covariancetfp.experimental.substrates.numpy.stats.correlationtfp.experimental.substrates.numpy.stats.count_integerstfp.experimental.substrates.numpy.stats.covariancetfp.experimental.substrates.numpy.stats.cumulative_variancetfp.experimental.substrates.numpy.stats.expected_calibration_errortfp.experimental.substrates.numpy.stats.expected_calibration_error_quantilestfp.experimental.substrates.numpy.stats.find_binstfp.experimental.substrates.numpy.stats.histogramtfp.experimental.substrates.numpy.stats.iterative_mergesorttfp.experimental.substrates.numpy.stats.kendalls_tautfp.experimental.substrates.numpy.stats.lexicographical_indirect_sorttfp.experimental.substrates.numpy.stats.log_average_probstfp.experimental.substrates.numpy.stats.log_loomean_exptfp.experimental.substrates.numpy.stats.log_loosum_exptfp.experimental.substrates.numpy.stats.log_soomean_exptfp.experimental.substrates.numpy.stats.log_soosum_exptfp.experimental.substrates.numpy.stats.moving_mean_variance_zero_debiasedtfp.experimental.substrates.numpy.stats.percentiletfp.experimental.substrates.numpy.stats.quantile_auctfp.experimental.substrates.numpy.stats.quantilestfp.experimental.substrates.numpy.stats.stddevtfp.experimental.substrates.numpy.stats.variancetfp.experimental.substrates.numpy.stats.windowed_meantfp.experimental.substrates.numpy.stats.windowed_variancetfp.experimental.substrates.numpy.utiltfp.experimental.substrates.numpy.util.DeferredTensortfp.experimental.substrates.numpy.util.ParameterPropertiestfp.experimental.substrates.numpy.util.SeedStreamtfp.experimental.substrates.numpy.util.TransformedVariabletfp.experimental.utiltfp.experimental.util.DeferredModuletfp.experimental.util.JitPublicMethodstfp.experimental.util.make_trainabletfp.experimental.vitfp.experimental.vi.build_affine_surrogate_posteriortfp.experimental.vi.build_affine_surrogate_posterior_from_base_distributiontfp.experimental.vi.build_asvi_surrogate_posteriortfp.experimental.vi.build_factored_surrogate_posteriortfp.experimental.vi.build_split_flow_surrogate_posteriortfp.experimental.vi.build_trainable_location_scale_distributiontfp.experimental.vi.register_asvi_substitution_ruletfp.experimental.vi.utiltfp.experimental.vi.util.build_linear_operator_zerostfp.experimental.vi.util.build_trainable_linear_operator_blocktfp.experimental.vi.util.build_trainable_linear_operator_diagtfp.experimental.vi.util.build_trainable_linear_operator_full_matrixtfp.experimental.vi.util.build_trainable_linear_operator_triltfp.glmtfp.glm.Bernoullitfp.glm.BernoulliNormalCDFtfp.glm.Binomialtfp.glm.CustomExponentialFamilytfp.glm.ExponentialFamilytfp.glm.GammaExptfp.glm.GammaSoftplustfp.glm.LogNormaltfp.glm.LogNormalSoftplustfp.glm.NegativeBinomialtfp.glm.NegativeBinomialSoftplustfp.glm.Normaltfp.glm.NormalReciprocaltfp.glm.Poissontfp.glm.PoissonSoftplustfp.glm.compute_predicted_linear_responsetfp.glm.convergence_criteria_small_relative_norm_weights_changetfp.glm.fittfp.glm.fit_one_steptfp.glm.fit_sparsetfp.glm.fit_sparse_one_steptfp.layerstfp.layers.AutoregressiveTransformtfp.layers.BlockwiseInitializertfp.layers.CategoricalMixtureOfOneHotCategoricaltfp.layers.Convolution1DFlipouttfp.layers.Convolution1DReparameterizationtfp.layers.Convolution2DFlipouttfp.layers.Convolution2DReparameterizationtfp.layers.Convolution3DFlipouttfp.layers.Convolution3DReparameterizationtfp.layers.DenseFlipouttfp.layers.DenseLocalReparameterizationtfp.layers.DenseReparameterizationtfp.layers.DenseVariationaltfp.layers.DistributionLambdatfp.layers.IndependentBernoullitfp.layers.IndependentLogistictfp.layers.IndependentNormaltfp.layers.IndependentPoissontfp.layers.KLDivergenceAddLosstfp.layers.KLDivergenceRegularizertfp.layers.MixtureLogistictfp.layers.MixtureNormaltfp.layers.MixtureSameFamilytfp.layers.MultivariateNormalTriLtfp.layers.OneHotCategoricaltfp.layers.VariableLayertfp.layers.VariationalGaussianProcesstfp.layers.conv_variationaltfp.layers.conv_variational.Convolution1DFlipouttfp.layers.conv_variational.Convolution1DReparameterizationtfp.layers.conv_variational.Convolution2DFlipouttfp.layers.conv_variational.Convolution2DReparameterizationtfp.layers.conv_variational.Convolution3DFlipouttfp.layers.conv_variational.Convolution3DReparameterizationtfp.layers.default_loc_scale_fntfp.layers.default_mean_field_normal_fntfp.layers.default_multivariate_normal_fntfp.layers.dense_variationaltfp.layers.dense_variational.DenseFlipouttfp.layers.dense_variational.DenseLocalReparameterizationtfp.layers.dense_variational.DenseReparameterizationtfp.layers.dense_variational_v2tfp.layers.dense_variational_v2.DenseVariationaltfp.layers.dense_variational_v2.kullback_leiblertfp.layers.dense_variational_v2.kullback_leibler.RegisterKLtfp.layers.dense_variational_v2.kullback_leibler.augment_kl_xent_docstfp.layers.dense_variational_v2.kullback_leibler.kl_divergencetfp.layers.distribution_layertfp.layers.distribution_layer.CategoricalMixtureOfOneHotCategoricaltfp.layers.distribution_layer.DistributionLambdatfp.layers.distribution_layer.IndependentBernoullitfp.layers.distribution_layer.IndependentLogistictfp.layers.distribution_layer.IndependentNormaltfp.layers.distribution_layer.IndependentPoissontfp.layers.distribution_layer.KLDivergenceAddLosstfp.layers.distribution_layer.KLDivergenceRegularizertfp.layers.distribution_layer.MixtureLogistictfp.layers.distribution_layer.MixtureNormaltfp.layers.distribution_layer.MixtureSameFamilytfp.layers.distribution_layer.MultivariateNormalTriLtfp.layers.distribution_layer.OneHotCategoricaltfp.layers.distribution_layer.VariationalGaussianProcesstfp.layers.initializerstfp.layers.initializers.BlockwiseInitializertfp.layers.masked_autoregressivetfp.layers.masked_autoregressive.AutoregressiveTransformtfp.layers.utiltfp.layers.util.default_loc_scale_fntfp.layers.util.default_mean_field_normal_fntfp.layers.util.default_multivariate_normal_fntfp.layers.util.deserialize_functiontfp.layers.util.serialize_functiontfp.layers.variable_inputtfp.layers.variable_input.VariableLayertfp.layers.weight_normtfp.layers.weight_norm.WeightNormtfp.mathtfp.math.MinimizeTraceableQuantitiestfp.math.atan_differencetfp.math.batch_interp_regular_1d_gridtfp.math.batch_interp_regular_nd_gridtfp.math.bessel_iv_ratiotfp.math.bessel_ivetfp.math.bessel_kvetfp.math.bracket_roottfp.math.cholesky_concattfp.math.cholesky_updatetfp.math.clip_by_value_preserve_gradienttfp.math.custom_gradienttfp.math.dawsntfp.math.dense_to_sparsetfp.math.diag_jacobiantfp.math.erfcinvtfp.math.erfcxtfp.math.fill_triangulartfp.math.fill_triangular_inversetfp.math.find_root_chandrupatlatfp.math.find_root_secanttfp.math.gram_schmidttfp.math.hypergeometrictfp.math.hypergeometric.hyp2f1_small_argumenttfp.math.igammacinvtfp.math.igammainvtfp.math.interp_regular_1d_gridtfp.math.lambertwtfp.math.lambertw_winitzki_approxtfp.math.lbetatfp.math.log1mexptfp.math.log1psquaretfp.math.log_add_exptfp.math.log_bessel_ivetfp.math.log_bessel_kvetfp.math.log_combinationstfp.math.log_coshtfp.math.log_cumsum_exptfp.math.log_gamma_correctiontfp.math.log_gamma_differencetfp.math.log_sub_exptfp.math.logerfctfp.math.logerfcxtfp.math.lu_matrix_inversetfp.math.lu_reconstructtfp.math.lu_solvetfp.math.minimizetfp.math.odetfp.math.ode.BDFtfp.math.ode.ChosenBySolvertfp.math.ode.Diagnosticstfp.math.ode.DormandPrincetfp.math.ode.Resultstfp.math.ode.Solvertfp.math.owens_ttfp.math.pivoted_choleskytfp.math.psd_kernelstfp.math.psd_kernels.ExpSinSquaredtfp.math.psd_kernels.ExponentiatedQuadratictfp.math.psd_kernels.FeatureScaledtfp.math.psd_kernels.FeatureTransformedtfp.math.psd_kernels.GeneralizedMaterntfp.math.psd_kernels.KumaraswamyTransformedtfp.math.psd_kernels.Lineartfp.math.psd_kernels.MaternFiveHalvestfp.math.psd_kernels.MaternOneHalftfp.math.psd_kernels.MaternThreeHalvestfp.math.psd_kernels.Parabolictfp.math.psd_kernels.Polynomialtfp.math.psd_kernels.PositiveSemidefiniteKerneltfp.math.psd_kernels.RationalQuadratictfp.math.psd_kernels.SchurComplementtfp.math.random_rademachertfp.math.random_rayleightfp.math.reduce_kahan_sumtfp.math.reduce_log_harmonic_mean_exptfp.math.reduce_logmeanexptfp.math.reduce_weighted_logsumexptfp.math.round_exponential_bump_functiontfp.math.scan_associativetfp.math.secant_roottfp.math.smoothersteptfp.math.soft_sorting_matrixtfp.math.soft_thresholdtfp.math.softplus_inversetfp.math.sparse_or_dense_matmultfp.math.sparse_or_dense_matvecmultfp.math.sqrt1pm1tfp.math.trapztfp.math.value_and_gradienttfp.mcmctfp.mcmc.CheckpointableStatesAndTracetfp.mcmc.DualAveragingStepSizeAdaptationtfp.mcmc.HamiltonianMonteCarlotfp.mcmc.MetropolisAdjustedLangevinAlgorithmtfp.mcmc.MetropolisHastingstfp.mcmc.NoUTurnSamplertfp.mcmc.RandomWalkMetropolistfp.mcmc.ReplicaExchangeMCtfp.mcmc.SimpleStepSizeAdaptationtfp.mcmc.SliceSamplertfp.mcmc.StatesAndTracetfp.mcmc.TransformedTransitionKerneltfp.mcmc.TransitionKerneltfp.mcmc.UncalibratedHamiltonianMonteCarlotfp.mcmc.UncalibratedLangevintfp.mcmc.UncalibratedRandomWalktfp.mcmc.default_swap_proposal_fntfp.mcmc.effective_sample_sizetfp.mcmc.even_odd_swap_proposal_fntfp.mcmc.make_simple_step_size_update_policytfp.mcmc.potential_scale_reductiontfp.mcmc.random_walk_normal_fntfp.mcmc.random_walk_uniform_fntfp.mcmc.sample_annealed_importance_chaintfp.mcmc.sample_chaintfp.mcmc.sample_halton_sequencetfp.monte_carlotfp.monte_carlo.expectationtfp.optimizertfp.optimizer.StochasticGradientLangevinDynamicstfp.optimizer.VariationalSGDtfp.optimizer.bfgs_minimizetfp.optimizer.converged_alltfp.optimizer.converged_anytfp.optimizer.convergence_criteriatfp.optimizer.convergence_criteria.ConvergenceCriteriontfp.optimizer.convergence_criteria.LossNotDecreasingtfp.optimizer.convergence_criteria.SuccessiveGradientsAreUncorrelatedtfp.optimizer.differential_evolution_minimizetfp.optimizer.differential_evolution_one_steptfp.optimizer.lbfgs_minimizetfp.optimizer.linesearchtfp.optimizer.linesearch.hager_zhangtfp.optimizer.nelder_mead_minimizetfp.optimizer.nelder_mead_one_steptfp.optimizer.proximal_hessian_sparse_minimizetfp.optimizer.proximal_hessian_sparse_one_steptfp.randomtfp.random.rademachertfp.random.rayleightfp.random.sanitize_seedtfp.random.spherical_uniformtfp.random.split_seedtfp.statstfp.stats.assign_log_moving_mean_exptfp.stats.assign_moving_mean_variancetfp.stats.auto_correlationtfp.stats.brier_decompositiontfp.stats.brier_scoretfp.stats.cholesky_covariancetfp.stats.correlationtfp.stats.count_integerstfp.stats.covariancetfp.stats.cumulative_variancetfp.stats.expected_calibration_errortfp.stats.expected_calibration_error_quantilestfp.stats.find_binstfp.stats.histogramtfp.stats.iterative_mergesorttfp.stats.kendalls_tautfp.stats.lexicographical_indirect_sorttfp.stats.log_average_probstfp.stats.log_loomean_exptfp.stats.log_loosum_exptfp.stats.log_soomean_exptfp.stats.log_soosum_exptfp.stats.moving_mean_variance_zero_debiasedtfp.stats.percentiletfp.stats.quantile_auctfp.stats.quantilestfp.stats.stddevtfp.stats.variancetfp.stats.windowed_meantfp.stats.windowed_variancetfp.ststfp.sts.AdditiveStateSpaceModeltfp.sts.Autoregressivetfp.sts.AutoregressiveStateSpaceModeltfp.sts.ConstrainedSeasonalStateSpaceModeltfp.sts.DynamicLinearRegressiontfp.sts.DynamicLinearRegressionStateSpaceModeltfp.sts.LinearRegressiontfp.sts.LocalLeveltfp.sts.LocalLevelStateSpaceModeltfp.sts.LocalLinearTrendtfp.sts.LocalLinearTrendStateSpaceModeltfp.sts.MaskedTimeSeriestfp.sts.Seasonaltfp.sts.SeasonalStateSpaceModeltfp.sts.SemiLocalLinearTrendtfp.sts.SemiLocalLinearTrendStateSpaceModeltfp.sts.SmoothSeasonaltfp.sts.SmoothSeasonalStateSpaceModeltfp.sts.SparseLinearRegressiontfp.sts.StructuralTimeSeriestfp.sts.Sumtfp.sts.build_factored_surrogate_posteriortfp.sts.decompose_by_componenttfp.sts.decompose_forecast_by_componenttfp.sts.fit_with_hmctfp.sts.forecasttfp.sts.impute_missing_valuestfp.sts.one_step_predictivetfp.sts.sample_uniform_initial_statetfp.substratestfp.substrates.jaxtfp.substrates.jax.bijectorstfp.substrates.jax.bijectors.AbsoluteValuetfp.substrates.jax.bijectors.Affinetfp.substrates.jax.bijectors.AffineLinearOperatortfp.substrates.jax.bijectors.AffineScalartfp.substrates.jax.bijectors.Ascendingtfp.substrates.jax.bijectors.AutoCompositeTensorBijectortfp.substrates.jax.bijectors.AutoregressiveNetworktfp.substrates.jax.bijectors.Bijectortfp.substrates.jax.bijectors.Blockwisetfp.substrates.jax.bijectors.Chaintfp.substrates.jax.bijectors.CholeskyOuterProducttfp.substrates.jax.bijectors.CholeskyToInvCholeskytfp.substrates.jax.bijectors.Compositiontfp.substrates.jax.bijectors.CorrelationCholeskytfp.substrates.jax.bijectors.Cumsumtfp.substrates.jax.bijectors.Exptfp.substrates.jax.bijectors.Expm1tfp.substrates.jax.bijectors.FillScaleTriLtfp.substrates.jax.bijectors.FillTriangulartfp.substrates.jax.bijectors.FrechetCDFtfp.substrates.jax.bijectors.GeneralizedExtremeValueCDFtfp.substrates.jax.bijectors.GeneralizedParetotfp.substrates.jax.bijectors.GompertzCDFtfp.substrates.jax.bijectors.GumbelCDFtfp.substrates.jax.bijectors.Identitytfp.substrates.jax.bijectors.Inlinetfp.substrates.jax.bijectors.Inverttfp.substrates.jax.bijectors.IteratedSigmoidCenteredtfp.substrates.jax.bijectors.JointMaptfp.substrates.jax.bijectors.KumaraswamyCDFtfp.substrates.jax.bijectors.LambertWTailtfp.substrates.jax.bijectors.Logtfp.substrates.jax.bijectors.Log1ptfp.substrates.jax.bijectors.MaskedAutoregressiveFlowtfp.substrates.jax.bijectors.MatrixInverseTriLtfp.substrates.jax.bijectors.MatvecLUtfp.substrates.jax.bijectors.MoyalCDFtfp.substrates.jax.bijectors.NormalCDFtfp.substrates.jax.bijectors.Orderedtfp.substrates.jax.bijectors.Padtfp.substrates.jax.bijectors.Permutetfp.substrates.jax.bijectors.Powertfp.substrates.jax.bijectors.PowerTransformtfp.substrates.jax.bijectors.RationalQuadraticSplinetfp.substrates.jax.bijectors.RayleighCDFtfp.substrates.jax.bijectors.RealNVPtfp.substrates.jax.bijectors.Reciprocaltfp.substrates.jax.bijectors.Reshapetfp.substrates.jax.bijectors.Restructuretfp.substrates.jax.bijectors.Scaletfp.substrates.jax.bijectors.ScaleMatvecDiagtfp.substrates.jax.bijectors.ScaleMatvecLUtfp.substrates.jax.bijectors.ScaleMatvecLinearOperatortfp.substrates.jax.bijectors.ScaleMatvecLinearOperatorBlocktfp.substrates.jax.bijectors.ScaleMatvecTriLtfp.substrates.jax.bijectors.Shifttfp.substrates.jax.bijectors.ShiftedGompertzCDFtfp.substrates.jax.bijectors.Sigmoidtfp.substrates.jax.bijectors.Sinhtfp.substrates.jax.bijectors.SinhArcsinhtfp.substrates.jax.bijectors.SoftCliptfp.substrates.jax.bijectors.Softfloortfp.substrates.jax.bijectors.SoftmaxCenteredtfp.substrates.jax.bijectors.Softplustfp.substrates.jax.bijectors.Softsigntfp.substrates.jax.bijectors.Splittfp.substrates.jax.bijectors.Squaretfp.substrates.jax.bijectors.Tanhtfp.substrates.jax.bijectors.TransformDiagonaltfp.substrates.jax.bijectors.Transposetfp.substrates.jax.bijectors.WeibullCDFtfp.substrates.jax.bijectors.masked_autoregressive_default_templatetfp.substrates.jax.bijectors.masked_densetfp.substrates.jax.bijectors.pack_sequence_astfp.substrates.jax.bijectors.real_nvp_default_templatetfp.substrates.jax.bijectors.tree_flattentfp.substrates.jax.distributionstfp.substrates.jax.distributions.Autoregressivetfp.substrates.jax.distributions.BatchBroadcasttfp.substrates.jax.distributions.BatchReshapetfp.substrates.jax.distributions.Batestfp.substrates.jax.distributions.Bernoullitfp.substrates.jax.distributions.Betatfp.substrates.jax.distributions.BetaBinomialtfp.substrates.jax.distributions.BetaQuotienttfp.substrates.jax.distributions.Binomialtfp.substrates.jax.distributions.Blockwisetfp.substrates.jax.distributions.Categoricaltfp.substrates.jax.distributions.Cauchytfp.substrates.jax.distributions.Chitfp.substrates.jax.distributions.Chi2tfp.substrates.jax.distributions.CholeskyLKJtfp.substrates.jax.distributions.DeterminantalPointProcesstfp.substrates.jax.distributions.Deterministictfp.substrates.jax.distributions.Dirichlettfp.substrates.jax.distributions.DirichletMultinomialtfp.substrates.jax.distributions.Distributiontfp.substrates.jax.distributions.DoublesidedMaxwelltfp.substrates.jax.distributions.Empiricaltfp.substrates.jax.distributions.ExpGammatfp.substrates.jax.distributions.ExpInverseGammatfp.substrates.jax.distributions.ExpRelaxedOneHotCategoricaltfp.substrates.jax.distributions.Exponentialtfp.substrates.jax.distributions.ExponentiallyModifiedGaussiantfp.substrates.jax.distributions.Gammatfp.substrates.jax.distributions.GammaGammatfp.substrates.jax.distributions.GaussianProcesstfp.substrates.jax.distributions.GaussianProcessRegressionModeltfp.substrates.jax.distributions.GeneralizedExtremeValuetfp.substrates.jax.distributions.GeneralizedNormaltfp.substrates.jax.distributions.GeneralizedParetotfp.substrates.jax.distributions.Geometrictfp.substrates.jax.distributions.Gumbeltfp.substrates.jax.distributions.HalfCauchytfp.substrates.jax.distributions.HalfNormaltfp.substrates.jax.distributions.HalfStudentTtfp.substrates.jax.distributions.HiddenMarkovModeltfp.substrates.jax.distributions.Horseshoetfp.substrates.jax.distributions.Independenttfp.substrates.jax.distributions.InverseGammatfp.substrates.jax.distributions.InverseGaussiantfp.substrates.jax.distributions.JohnsonSUtfp.substrates.jax.distributions.JointDistributiontfp.substrates.jax.distributions.JointDistribution.Roottfp.substrates.jax.distributions.JointDistributionCoroutinetfp.substrates.jax.distributions.JointDistributionCoroutine.Roottfp.substrates.jax.distributions.JointDistributionCoroutineAutoBatchedtfp.substrates.jax.distributions.JointDistributionCoroutineAutoBatched.Roottfp.substrates.jax.distributions.JointDistributionNamedtfp.substrates.jax.distributions.JointDistributionNamed.Roottfp.substrates.jax.distributions.JointDistributionNamedAutoBatchedtfp.substrates.jax.distributions.JointDistributionNamedAutoBatched.Roottfp.substrates.jax.distributions.JointDistributionSequentialtfp.substrates.jax.distributions.JointDistributionSequential.Roottfp.substrates.jax.distributions.JointDistributionSequentialAutoBatchedtfp.substrates.jax.distributions.JointDistributionSequentialAutoBatched.Roottfp.substrates.jax.distributions.Kumaraswamytfp.substrates.jax.distributions.LKJtfp.substrates.jax.distributions.LambertWDistributiontfp.substrates.jax.distributions.LambertWNormaltfp.substrates.jax.distributions.Laplacetfp.substrates.jax.distributions.LinearGaussianStateSpaceModeltfp.substrates.jax.distributions.LogLogistictfp.substrates.jax.distributions.LogNormaltfp.substrates.jax.distributions.Logistictfp.substrates.jax.distributions.LogitNormaltfp.substrates.jax.distributions.Maskedtfp.substrates.jax.distributions.MatrixNormalLinearOperatortfp.substrates.jax.distributions.MatrixTLinearOperatortfp.substrates.jax.distributions.MixtureSameFamilytfp.substrates.jax.distributions.Moyaltfp.substrates.jax.distributions.Multinomialtfp.substrates.jax.distributions.MultivariateNormalDiagtfp.substrates.jax.distributions.MultivariateNormalDiagPlusLowRanktfp.substrates.jax.distributions.MultivariateNormalFullCovariancetfp.substrates.jax.distributions.MultivariateNormalLinearOperatortfp.substrates.jax.distributions.MultivariateNormalTriLtfp.substrates.jax.distributions.MultivariateStudentTLinearOperatortfp.substrates.jax.distributions.NegativeBinomialtfp.substrates.jax.distributions.Normaltfp.substrates.jax.distributions.NormalInverseGaussiantfp.substrates.jax.distributions.OneHotCategoricaltfp.substrates.jax.distributions.OrderedLogistictfp.substrates.jax.distributions.PERTtfp.substrates.jax.distributions.Paretotfp.substrates.jax.distributions.PlackettLucetfp.substrates.jax.distributions.Poissontfp.substrates.jax.distributions.PoissonLogNormalQuadratureCompoundtfp.substrates.jax.distributions.PowerSphericaltfp.substrates.jax.distributions.ProbitBernoullitfp.substrates.jax.distributions.QuantizedDistributiontfp.substrates.jax.distributions.RegisterKLtfp.substrates.jax.distributions.RelaxedBernoullitfp.substrates.jax.distributions.RelaxedOneHotCategoricaltfp.substrates.jax.distributions.ReparameterizationTypetfp.substrates.jax.distributions.Sampletfp.substrates.jax.distributions.SigmoidBetatfp.substrates.jax.distributions.SinhArcsinhtfp.substrates.jax.distributions.Skellamtfp.substrates.jax.distributions.SphericalUniformtfp.substrates.jax.distributions.StoppingRatioLogistictfp.substrates.jax.distributions.StudentTtfp.substrates.jax.distributions.StudentTProcesstfp.substrates.jax.distributions.TransformedDistributiontfp.substrates.jax.distributions.Triangulartfp.substrates.jax.distributions.TruncatedCauchytfp.substrates.jax.distributions.TruncatedNormaltfp.substrates.jax.distributions.Uniformtfp.substrates.jax.distributions.VariationalGaussianProcesstfp.substrates.jax.distributions.VectorDeterministictfp.substrates.jax.distributions.VectorExponentialDiagtfp.substrates.jax.distributions.VonMisestfp.substrates.jax.distributions.VonMisesFishertfp.substrates.jax.distributions.Weibulltfp.substrates.jax.distributions.WishartLinearOperatortfp.substrates.jax.distributions.WishartTriLtfp.substrates.jax.distributions.Zipftfp.substrates.jax.distributions.independent_joint_distribution_from_structuretfp.substrates.jax.distributions.kl_divergencetfp.substrates.jax.distributions.mvn_conjugate_linear_updatetfp.substrates.jax.distributions.normal_conjugates_known_scale_posteriortfp.substrates.jax.distributions.normal_conjugates_known_scale_predictivetfp.substrates.jax.distributions.quadrature_scheme_lognormal_gauss_hermitetfp.substrates.jax.distributions.quadrature_scheme_lognormal_quantilestfp.substrates.jax.mathtfp.substrates.jax.math.atan_differencetfp.substrates.jax.math.batch_interp_regular_1d_gridtfp.substrates.jax.math.batch_interp_regular_nd_gridtfp.substrates.jax.math.bessel_iv_ratiotfp.substrates.jax.math.bessel_ivetfp.substrates.jax.math.bessel_kvetfp.substrates.jax.math.bracket_roottfp.substrates.jax.math.cholesky_concattfp.substrates.jax.math.cholesky_updatetfp.substrates.jax.math.clip_by_value_preserve_gradienttfp.substrates.jax.math.custom_gradienttfp.substrates.jax.math.dawsntfp.substrates.jax.math.diag_jacobiantfp.substrates.jax.math.erfcinvtfp.substrates.jax.math.erfcxtfp.substrates.jax.math.fill_triangulartfp.substrates.jax.math.fill_triangular_inversetfp.substrates.jax.math.find_root_chandrupatlatfp.substrates.jax.math.find_root_secanttfp.substrates.jax.math.gram_schmidttfp.substrates.jax.math.hypergeometrictfp.substrates.jax.math.hypergeometric.hyp2f1_small_argumenttfp.substrates.jax.math.igammacinvtfp.substrates.jax.math.igammainvtfp.substrates.jax.math.interp_regular_1d_gridtfp.substrates.jax.math.lambertwtfp.substrates.jax.math.lambertw_winitzki_approxtfp.substrates.jax.math.lbetatfp.substrates.jax.math.log1mexptfp.substrates.jax.math.log1psquaretfp.substrates.jax.math.log_add_exptfp.substrates.jax.math.log_bessel_ivetfp.substrates.jax.math.log_bessel_kvetfp.substrates.jax.math.log_combinationstfp.substrates.jax.math.log_coshtfp.substrates.jax.math.log_cumsum_exptfp.substrates.jax.math.log_gamma_correctiontfp.substrates.jax.math.log_gamma_differencetfp.substrates.jax.math.log_sub_exptfp.substrates.jax.math.logerfctfp.substrates.jax.math.logerfcxtfp.substrates.jax.math.lu_matrix_inversetfp.substrates.jax.math.lu_reconstructtfp.substrates.jax.math.lu_solvetfp.substrates.jax.math.owens_ttfp.substrates.jax.math.pivoted_choleskytfp.substrates.jax.math.psd_kernelstfp.substrates.jax.math.psd_kernels.ExpSinSquaredtfp.substrates.jax.math.psd_kernels.ExponentiatedQuadratictfp.substrates.jax.math.psd_kernels.FeatureScaledtfp.substrates.jax.math.psd_kernels.FeatureTransformedtfp.substrates.jax.math.psd_kernels.GeneralizedMaterntfp.substrates.jax.math.psd_kernels.KumaraswamyTransformedtfp.substrates.jax.math.psd_kernels.Lineartfp.substrates.jax.math.psd_kernels.MaternFiveHalvestfp.substrates.jax.math.psd_kernels.MaternOneHalftfp.substrates.jax.math.psd_kernels.MaternThreeHalvestfp.substrates.jax.math.psd_kernels.Parabolictfp.substrates.jax.math.psd_kernels.Polynomialtfp.substrates.jax.math.psd_kernels.PositiveSemidefiniteKerneltfp.substrates.jax.math.psd_kernels.RationalQuadratictfp.substrates.jax.math.psd_kernels.SchurComplementtfp.substrates.jax.math.random_rademachertfp.substrates.jax.math.random_rayleightfp.substrates.jax.math.reduce_kahan_sumtfp.substrates.jax.math.reduce_log_harmonic_mean_exptfp.substrates.jax.math.reduce_logmeanexptfp.substrates.jax.math.reduce_weighted_logsumexptfp.substrates.jax.math.round_exponential_bump_functiontfp.substrates.jax.math.scan_associativetfp.substrates.jax.math.secant_roottfp.substrates.jax.math.smoothersteptfp.substrates.jax.math.soft_sorting_matrixtfp.substrates.jax.math.soft_thresholdtfp.substrates.jax.math.softplus_inversetfp.substrates.jax.math.sparse_or_dense_matmultfp.substrates.jax.math.sparse_or_dense_matvecmultfp.substrates.jax.math.sqrt1pm1tfp.substrates.jax.math.trapztfp.substrates.jax.math.value_and_gradienttfp.substrates.jax.mcmctfp.substrates.jax.mcmc.CheckpointableStatesAndTracetfp.substrates.jax.mcmc.DualAveragingStepSizeAdaptationtfp.substrates.jax.mcmc.HamiltonianMonteCarlotfp.substrates.jax.mcmc.MetropolisAdjustedLangevinAlgorithmtfp.substrates.jax.mcmc.MetropolisHastingstfp.substrates.jax.mcmc.NoUTurnSamplertfp.substrates.jax.mcmc.RandomWalkMetropolistfp.substrates.jax.mcmc.ReplicaExchangeMCtfp.substrates.jax.mcmc.SimpleStepSizeAdaptationtfp.substrates.jax.mcmc.SliceSamplertfp.substrates.jax.mcmc.StatesAndTracetfp.substrates.jax.mcmc.TransformedTransitionKerneltfp.substrates.jax.mcmc.TransitionKerneltfp.substrates.jax.mcmc.UncalibratedHamiltonianMonteCarlotfp.substrates.jax.mcmc.UncalibratedLangevintfp.substrates.jax.mcmc.UncalibratedRandomWalktfp.substrates.jax.mcmc.default_swap_proposal_fntfp.substrates.jax.mcmc.effective_sample_sizetfp.substrates.jax.mcmc.even_odd_swap_proposal_fntfp.substrates.jax.mcmc.make_simple_step_size_update_policytfp.substrates.jax.mcmc.potential_scale_reductiontfp.substrates.jax.mcmc.random_walk_normal_fntfp.substrates.jax.mcmc.random_walk_uniform_fntfp.substrates.jax.mcmc.sample_annealed_importance_chaintfp.substrates.jax.mcmc.sample_chaintfp.substrates.jax.mcmc.sample_halton_sequencetfp.substrates.jax.optimizertfp.substrates.jax.optimizer.bfgs_minimizetfp.substrates.jax.optimizer.converged_alltfp.substrates.jax.optimizer.converged_anytfp.substrates.jax.optimizer.lbfgs_minimizetfp.substrates.jax.optimizer.linesearchtfp.substrates.jax.optimizer.linesearch.hager_zhangtfp.substrates.jax.optimizer.nelder_mead_minimizetfp.substrates.jax.optimizer.nelder_mead_one_steptfp.substrates.jax.randomtfp.substrates.jax.random.rademachertfp.substrates.jax.random.rayleightfp.substrates.jax.random.sanitize_seedtfp.substrates.jax.random.spherical_uniformtfp.substrates.jax.random.split_seedtfp.substrates.jax.statstfp.substrates.jax.stats.assign_log_moving_mean_exptfp.substrates.jax.stats.assign_moving_mean_variancetfp.substrates.jax.stats.auto_correlationtfp.substrates.jax.stats.brier_decompositiontfp.substrates.jax.stats.brier_scoretfp.substrates.jax.stats.cholesky_covariancetfp.substrates.jax.stats.correlationtfp.substrates.jax.stats.count_integerstfp.substrates.jax.stats.covariancetfp.substrates.jax.stats.cumulative_variancetfp.substrates.jax.stats.expected_calibration_errortfp.substrates.jax.stats.expected_calibration_error_quantilestfp.substrates.jax.stats.find_binstfp.substrates.jax.stats.histogramtfp.substrates.jax.stats.iterative_mergesorttfp.substrates.jax.stats.kendalls_tautfp.substrates.jax.stats.lexicographical_indirect_sorttfp.substrates.jax.stats.log_average_probstfp.substrates.jax.stats.log_loomean_exptfp.substrates.jax.stats.log_loosum_exptfp.substrates.jax.stats.log_soomean_exptfp.substrates.jax.stats.log_soosum_exptfp.substrates.jax.stats.moving_mean_variance_zero_debiasedtfp.substrates.jax.stats.percentiletfp.substrates.jax.stats.quantilestfp.substrates.jax.stats.stddevtfp.substrates.jax.stats.variancetfp.substrates.jax.stats.windowed_meantfp.substrates.jax.stats.windowed_variancetfp.substrates.jax.utiltfp.substrates.jax.util.DeferredTensortfp.substrates.jax.util.ParameterPropertiestfp.substrates.jax.util.SeedStreamtfp.substrates.jax.util.TransformedVariabletfp.substrates.numpytfp.substrates.numpy.bijectorstfp.substrates.numpy.bijectors.AbsoluteValuetfp.substrates.numpy.bijectors.Affinetfp.substrates.numpy.bijectors.AffineLinearOperatortfp.substrates.numpy.bijectors.AffineScalartfp.substrates.numpy.bijectors.Ascendingtfp.substrates.numpy.bijectors.AutoCompositeTensorBijectortfp.substrates.numpy.bijectors.AutoregressiveNetworktfp.substrates.numpy.bijectors.Bijectortfp.substrates.numpy.bijectors.Blockwisetfp.substrates.numpy.bijectors.Chaintfp.substrates.numpy.bijectors.CholeskyOuterProducttfp.substrates.numpy.bijectors.CholeskyToInvCholeskytfp.substrates.numpy.bijectors.Compositiontfp.substrates.numpy.bijectors.CorrelationCholeskytfp.substrates.numpy.bijectors.Cumsumtfp.substrates.numpy.bijectors.Exptfp.substrates.numpy.bijectors.Expm1tfp.substrates.numpy.bijectors.FillScaleTriLtfp.substrates.numpy.bijectors.FillTriangulartfp.substrates.numpy.bijectors.FrechetCDFtfp.substrates.numpy.bijectors.GeneralizedExtremeValueCDFtfp.substrates.numpy.bijectors.GeneralizedParetotfp.substrates.numpy.bijectors.GompertzCDFtfp.substrates.numpy.bijectors.GumbelCDFtfp.substrates.numpy.bijectors.Identitytfp.substrates.numpy.bijectors.Inlinetfp.substrates.numpy.bijectors.Inverttfp.substrates.numpy.bijectors.IteratedSigmoidCenteredtfp.substrates.numpy.bijectors.JointMaptfp.substrates.numpy.bijectors.KumaraswamyCDFtfp.substrates.numpy.bijectors.LambertWTailtfp.substrates.numpy.bijectors.Logtfp.substrates.numpy.bijectors.Log1ptfp.substrates.numpy.bijectors.MaskedAutoregressiveFlowtfp.substrates.numpy.bijectors.MatrixInverseTriLtfp.substrates.numpy.bijectors.MatvecLUtfp.substrates.numpy.bijectors.MoyalCDFtfp.substrates.numpy.bijectors.NormalCDFtfp.substrates.numpy.bijectors.Orderedtfp.substrates.numpy.bijectors.Padtfp.substrates.numpy.bijectors.Permutetfp.substrates.numpy.bijectors.Powertfp.substrates.numpy.bijectors.PowerTransformtfp.substrates.numpy.bijectors.RationalQuadraticSplinetfp.substrates.numpy.bijectors.RayleighCDFtfp.substrates.numpy.bijectors.RealNVPtfp.substrates.numpy.bijectors.Reciprocaltfp.substrates.numpy.bijectors.Reshapetfp.substrates.numpy.bijectors.Restructuretfp.substrates.numpy.bijectors.Scaletfp.substrates.numpy.bijectors.ScaleMatvecDiagtfp.substrates.numpy.bijectors.ScaleMatvecLUtfp.substrates.numpy.bijectors.ScaleMatvecLinearOperatortfp.substrates.numpy.bijectors.ScaleMatvecLinearOperatorBlocktfp.substrates.numpy.bijectors.ScaleMatvecTriLtfp.substrates.numpy.bijectors.Shifttfp.substrates.numpy.bijectors.ShiftedGompertzCDFtfp.substrates.numpy.bijectors.Sigmoidtfp.substrates.numpy.bijectors.Sinhtfp.substrates.numpy.bijectors.SinhArcsinhtfp.substrates.numpy.bijectors.SoftCliptfp.substrates.numpy.bijectors.Softfloortfp.substrates.numpy.bijectors.SoftmaxCenteredtfp.substrates.numpy.bijectors.Softplustfp.substrates.numpy.bijectors.Softsigntfp.substrates.numpy.bijectors.Splittfp.substrates.numpy.bijectors.Squaretfp.substrates.numpy.bijectors.Tanhtfp.substrates.numpy.bijectors.TransformDiagonaltfp.substrates.numpy.bijectors.Transposetfp.substrates.numpy.bijectors.WeibullCDFtfp.substrates.numpy.bijectors.masked_autoregressive_default_templatetfp.substrates.numpy.bijectors.masked_densetfp.substrates.numpy.bijectors.pack_sequence_astfp.substrates.numpy.bijectors.real_nvp_default_templatetfp.substrates.numpy.bijectors.tree_flattentfp.substrates.numpy.distributionstfp.substrates.numpy.distributions.Autoregressivetfp.substrates.numpy.distributions.BatchBroadcasttfp.substrates.numpy.distributions.BatchReshapetfp.substrates.numpy.distributions.Batestfp.substrates.numpy.distributions.Bernoullitfp.substrates.numpy.distributions.Betatfp.substrates.numpy.distributions.BetaBinomialtfp.substrates.numpy.distributions.BetaQuotienttfp.substrates.numpy.distributions.Binomialtfp.substrates.numpy.distributions.Blockwisetfp.substrates.numpy.distributions.Categoricaltfp.substrates.numpy.distributions.Cauchytfp.substrates.numpy.distributions.Chitfp.substrates.numpy.distributions.Chi2tfp.substrates.numpy.distributions.CholeskyLKJtfp.substrates.numpy.distributions.DeterminantalPointProcesstfp.substrates.numpy.distributions.Deterministictfp.substrates.numpy.distributions.Dirichlettfp.substrates.numpy.distributions.DirichletMultinomialtfp.substrates.numpy.distributions.Distributiontfp.substrates.numpy.distributions.DoublesidedMaxwelltfp.substrates.numpy.distributions.Empiricaltfp.substrates.numpy.distributions.ExpGammatfp.substrates.numpy.distributions.ExpInverseGammatfp.substrates.numpy.distributions.ExpRelaxedOneHotCategoricaltfp.substrates.numpy.distributions.Exponentialtfp.substrates.numpy.distributions.ExponentiallyModifiedGaussiantfp.substrates.numpy.distributions.Gammatfp.substrates.numpy.distributions.GammaGammatfp.substrates.numpy.distributions.GaussianProcesstfp.substrates.numpy.distributions.GaussianProcessRegressionModeltfp.substrates.numpy.distributions.GeneralizedExtremeValuetfp.substrates.numpy.distributions.GeneralizedNormaltfp.substrates.numpy.distributions.GeneralizedParetotfp.substrates.numpy.distributions.Geometrictfp.substrates.numpy.distributions.Gumbeltfp.substrates.numpy.distributions.HalfCauchytfp.substrates.numpy.distributions.HalfNormaltfp.substrates.numpy.distributions.HalfStudentTtfp.substrates.numpy.distributions.HiddenMarkovModeltfp.substrates.numpy.distributions.Horseshoetfp.substrates.numpy.distributions.Independenttfp.substrates.numpy.distributions.InverseGammatfp.substrates.numpy.distributions.InverseGaussiantfp.substrates.numpy.distributions.JohnsonSUtfp.substrates.numpy.distributions.JointDistributiontfp.substrates.numpy.distributions.JointDistribution.Roottfp.substrates.numpy.distributions.JointDistributionCoroutinetfp.substrates.numpy.distributions.JointDistributionCoroutine.Roottfp.substrates.numpy.distributions.JointDistributionCoroutineAutoBatchedtfp.substrates.numpy.distributions.JointDistributionCoroutineAutoBatched.Roottfp.substrates.numpy.distributions.JointDistributionNamedtfp.substrates.numpy.distributions.JointDistributionNamed.Roottfp.substrates.numpy.distributions.JointDistributionNamedAutoBatchedtfp.substrates.numpy.distributions.JointDistributionNamedAutoBatched.Roottfp.substrates.numpy.distributions.JointDistributionSequentialtfp.substrates.numpy.distributions.JointDistributionSequential.Roottfp.substrates.numpy.distributions.JointDistributionSequentialAutoBatchedtfp.substrates.numpy.distributions.JointDistributionSequentialAutoBatched.Roottfp.substrates.numpy.distributions.Kumaraswamytfp.substrates.numpy.distributions.LKJtfp.substrates.numpy.distributions.LambertWDistributiontfp.substrates.numpy.distributions.LambertWNormaltfp.substrates.numpy.distributions.Laplacetfp.substrates.numpy.distributions.LinearGaussianStateSpaceModeltfp.substrates.numpy.distributions.LogLogistictfp.substrates.numpy.distributions.LogNormaltfp.substrates.numpy.distributions.Logistictfp.substrates.numpy.distributions.LogitNormaltfp.substrates.numpy.distributions.Maskedtfp.substrates.numpy.distributions.MatrixNormalLinearOperatortfp.substrates.numpy.distributions.MatrixTLinearOperatortfp.substrates.numpy.distributions.MixtureSameFamilytfp.substrates.numpy.distributions.Moyaltfp.substrates.numpy.distributions.Multinomialtfp.substrates.numpy.distributions.MultivariateNormalDiagtfp.substrates.numpy.distributions.MultivariateNormalDiagPlusLowRanktfp.substrates.numpy.distributions.MultivariateNormalFullCovariancetfp.substrates.numpy.distributions.MultivariateNormalLinearOperatortfp.substrates.numpy.distributions.MultivariateNormalTriLtfp.substrates.numpy.distributions.MultivariateStudentTLinearOperatortfp.substrates.numpy.distributions.NegativeBinomialtfp.substrates.numpy.distributions.Normaltfp.substrates.numpy.distributions.NormalInverseGaussiantfp.substrates.numpy.distributions.OneHotCategoricaltfp.substrates.numpy.distributions.OrderedLogistictfp.substrates.numpy.distributions.PERTtfp.substrates.numpy.distributions.Paretotfp.substrates.numpy.distributions.PlackettLucetfp.substrates.numpy.distributions.Poissontfp.substrates.numpy.distributions.PoissonLogNormalQuadratureCompoundtfp.substrates.numpy.distributions.PowerSphericaltfp.substrates.numpy.distributions.ProbitBernoullitfp.substrates.numpy.distributions.QuantizedDistributiontfp.substrates.numpy.distributions.RegisterKLtfp.substrates.numpy.distributions.RelaxedBernoullitfp.substrates.numpy.distributions.RelaxedOneHotCategoricaltfp.substrates.numpy.distributions.ReparameterizationTypetfp.substrates.numpy.distributions.Sampletfp.substrates.numpy.distributions.SigmoidBetatfp.substrates.numpy.distributions.SinhArcsinhtfp.substrates.numpy.distributions.Skellamtfp.substrates.numpy.distributions.SphericalUniformtfp.substrates.numpy.distributions.StoppingRatioLogistictfp.substrates.numpy.distributions.StudentTtfp.substrates.numpy.distributions.StudentTProcesstfp.substrates.numpy.distributions.TransformedDistributiontfp.substrates.numpy.distributions.Triangulartfp.substrates.numpy.distributions.TruncatedCauchytfp.substrates.numpy.distributions.TruncatedNormaltfp.substrates.numpy.distributions.Uniformtfp.substrates.numpy.distributions.VariationalGaussianProcesstfp.substrates.numpy.distributions.VectorDeterministictfp.substrates.numpy.distributions.VectorExponentialDiagtfp.substrates.numpy.distributions.VonMisestfp.substrates.numpy.distributions.VonMisesFishertfp.substrates.numpy.distributions.Weibulltfp.substrates.numpy.distributions.WishartLinearOperatortfp.substrates.numpy.distributions.WishartTriLtfp.substrates.numpy.distributions.Zipftfp.substrates.numpy.distributions.independent_joint_distribution_from_structuretfp.substrates.numpy.distributions.kl_divergencetfp.substrates.numpy.distributions.mvn_conjugate_linear_updatetfp.substrates.numpy.distributions.normal_conjugates_known_scale_posteriortfp.substrates.numpy.distributions.normal_conjugates_known_scale_predictivetfp.substrates.numpy.distributions.quadrature_scheme_lognormal_gauss_hermitetfp.substrates.numpy.distributions.quadrature_scheme_lognormal_quantilestfp.substrates.numpy.mathtfp.substrates.numpy.math.atan_differencetfp.substrates.numpy.math.batch_interp_regular_1d_gridtfp.substrates.numpy.math.batch_interp_regular_nd_gridtfp.substrates.numpy.math.bessel_iv_ratiotfp.substrates.numpy.math.bessel_ivetfp.substrates.numpy.math.bessel_kvetfp.substrates.numpy.math.bracket_roottfp.substrates.numpy.math.cholesky_concattfp.substrates.numpy.math.cholesky_updatetfp.substrates.numpy.math.clip_by_value_preserve_gradienttfp.substrates.numpy.math.custom_gradienttfp.substrates.numpy.math.dawsntfp.substrates.numpy.math.diag_jacobiantfp.substrates.numpy.math.erfcinvtfp.substrates.numpy.math.erfcxtfp.substrates.numpy.math.fill_triangulartfp.substrates.numpy.math.fill_triangular_inversetfp.substrates.numpy.math.find_root_chandrupatlatfp.substrates.numpy.math.find_root_secanttfp.substrates.numpy.math.gram_schmidttfp.substrates.numpy.math.hypergeometrictfp.substrates.numpy.math.hypergeometric.hyp2f1_small_argumenttfp.substrates.numpy.math.igammacinvtfp.substrates.numpy.math.igammainvtfp.substrates.numpy.math.interp_regular_1d_gridtfp.substrates.numpy.math.lambertwtfp.substrates.numpy.math.lambertw_winitzki_approxtfp.substrates.numpy.math.lbetatfp.substrates.numpy.math.log1mexptfp.substrates.numpy.math.log1psquaretfp.substrates.numpy.math.log_add_exptfp.substrates.numpy.math.log_bessel_ivetfp.substrates.numpy.math.log_bessel_kvetfp.substrates.numpy.math.log_combinationstfp.substrates.numpy.math.log_coshtfp.substrates.numpy.math.log_cumsum_exptfp.substrates.numpy.math.log_gamma_correctiontfp.substrates.numpy.math.log_gamma_differencetfp.substrates.numpy.math.log_sub_exptfp.substrates.numpy.math.logerfctfp.substrates.numpy.math.logerfcxtfp.substrates.numpy.math.lu_matrix_inversetfp.substrates.numpy.math.lu_reconstructtfp.substrates.numpy.math.lu_solvetfp.substrates.numpy.math.owens_ttfp.substrates.numpy.math.pivoted_choleskytfp.substrates.numpy.math.psd_kernelstfp.substrates.numpy.math.psd_kernels.ExpSinSquaredtfp.substrates.numpy.math.psd_kernels.ExponentiatedQuadratictfp.substrates.numpy.math.psd_kernels.FeatureScaledtfp.substrates.numpy.math.psd_kernels.FeatureTransformedtfp.substrates.numpy.math.psd_kernels.GeneralizedMaterntfp.substrates.numpy.math.psd_kernels.KumaraswamyTransformedtfp.substrates.numpy.math.psd_kernels.Lineartfp.substrates.numpy.math.psd_kernels.MaternFiveHalvestfp.substrates.numpy.math.psd_kernels.MaternOneHalftfp.substrates.numpy.math.psd_kernels.MaternThreeHalvestfp.substrates.numpy.math.psd_kernels.Parabolictfp.substrates.numpy.math.psd_kernels.Polynomialtfp.substrates.numpy.math.psd_kernels.PositiveSemidefiniteKerneltfp.substrates.numpy.math.psd_kernels.RationalQuadratictfp.substrates.numpy.math.psd_kernels.SchurComplementtfp.substrates.numpy.math.random_rademachertfp.substrates.numpy.math.random_rayleightfp.substrates.numpy.math.reduce_kahan_sumtfp.substrates.numpy.math.reduce_log_harmonic_mean_exptfp.substrates.numpy.math.reduce_logmeanexptfp.substrates.numpy.math.reduce_weighted_logsumexptfp.substrates.numpy.math.round_exponential_bump_functiontfp.substrates.numpy.math.scan_associativetfp.substrates.numpy.math.secant_roottfp.substrates.numpy.math.smoothersteptfp.substrates.numpy.math.soft_sorting_matrixtfp.substrates.numpy.math.soft_thresholdtfp.substrates.numpy.math.softplus_inversetfp.substrates.numpy.math.sparse_or_dense_matmultfp.substrates.numpy.math.sparse_or_dense_matvecmultfp.substrates.numpy.math.sqrt1pm1tfp.substrates.numpy.math.trapztfp.substrates.numpy.math.value_and_gradienttfp.substrates.numpy.mcmctfp.substrates.numpy.mcmc.CheckpointableStatesAndTracetfp.substrates.numpy.mcmc.DualAveragingStepSizeAdaptationtfp.substrates.numpy.mcmc.MetropolisHastingstfp.substrates.numpy.mcmc.RandomWalkMetropolistfp.substrates.numpy.mcmc.ReplicaExchangeMCtfp.substrates.numpy.mcmc.SimpleStepSizeAdaptationtfp.substrates.numpy.mcmc.SliceSamplertfp.substrates.numpy.mcmc.StatesAndTracetfp.substrates.numpy.mcmc.TransformedTransitionKerneltfp.substrates.numpy.mcmc.TransitionKerneltfp.substrates.numpy.mcmc.UncalibratedRandomWalktfp.substrates.numpy.mcmc.default_swap_proposal_fntfp.substrates.numpy.mcmc.effective_sample_sizetfp.substrates.numpy.mcmc.even_odd_swap_proposal_fntfp.substrates.numpy.mcmc.potential_scale_reductiontfp.substrates.numpy.mcmc.random_walk_normal_fntfp.substrates.numpy.mcmc.random_walk_uniform_fntfp.substrates.numpy.mcmc.sample_annealed_importance_chaintfp.substrates.numpy.mcmc.sample_chaintfp.substrates.numpy.mcmc.sample_halton_sequencetfp.substrates.numpy.optimizertfp.substrates.numpy.optimizer.bfgs_minimizetfp.substrates.numpy.optimizer.converged_alltfp.substrates.numpy.optimizer.converged_anytfp.substrates.numpy.optimizer.lbfgs_minimizetfp.substrates.numpy.optimizer.linesearchtfp.substrates.numpy.optimizer.linesearch.hager_zhangtfp.substrates.numpy.optimizer.nelder_mead_minimizetfp.substrates.numpy.optimizer.nelder_mead_one_steptfp.substrates.numpy.randomtfp.substrates.numpy.random.rademachertfp.substrates.numpy.random.rayleightfp.substrates.numpy.random.sanitize_seedtfp.substrates.numpy.random.spherical_uniformtfp.substrates.numpy.random.split_seedtfp.substrates.numpy.statstfp.substrates.numpy.stats.assign_log_moving_mean_exptfp.substrates.numpy.stats.assign_moving_mean_variancetfp.substrates.numpy.stats.auto_correlationtfp.substrates.numpy.stats.brier_decompositiontfp.substrates.numpy.stats.brier_scoretfp.substrates.numpy.stats.cholesky_covariancetfp.substrates.numpy.stats.correlationtfp.substrates.numpy.stats.count_integerstfp.substrates.numpy.stats.covariancetfp.substrates.numpy.stats.cumulative_variancetfp.substrates.numpy.stats.expected_calibration_errortfp.substrates.numpy.stats.expected_calibration_error_quantilestfp.substrates.numpy.stats.find_binstfp.substrates.numpy.stats.histogramtfp.substrates.numpy.stats.iterative_mergesorttfp.substrates.numpy.stats.kendalls_tautfp.substrates.numpy.stats.lexicographical_indirect_sorttfp.substrates.numpy.stats.log_average_probstfp.substrates.numpy.stats.log_loomean_exptfp.substrates.numpy.stats.log_loosum_exptfp.substrates.numpy.stats.log_soomean_exptfp.substrates.numpy.stats.log_soosum_exptfp.substrates.numpy.stats.moving_mean_variance_zero_debiasedtfp.substrates.numpy.stats.percentiletfp.substrates.numpy.stats.quantile_auctfp.substrates.numpy.stats.quantilestfp.substrates.numpy.stats.stddevtfp.substrates.numpy.stats.variancetfp.substrates.numpy.stats.windowed_meantfp.substrates.numpy.stats.windowed_variancetfp.substrates.numpy.utiltfp.substrates.numpy.util.DeferredTensortfp.substrates.numpy.util.ParameterPropertiestfp.substrates.numpy.util.SeedStreamtfp.substrates.numpy.util.TransformedVariabletfp.utiltfp.util.DeferredTensortfp.util.ParameterPropertiestfp.util.SeedStreamtfp.util.TransformedVariabletfp.vitfp.vi.amari_alphatfp.vi.arithmetic_geometrictfp.vi.chi_squaretfp.vi.csiszar_vimcotfp.vi.dual_csiszar_functiontfp.vi.fit_surrogate_posteriortfp.vi.jeffreystfp.vi.jensen_shannontfp.vi.kl_forwardtfp.vi.kl_reversetfp.vi.log1p_abstfp.vi.modified_gantfp.vi.monte_carlo_variational_losstfp.vi.mutual_informationtfp.vi.mutual_information.lower_bound_barber_agakovtfp.vi.mutual_information.lower_bound_info_ncetfp.vi.mutual_information.lower_bound_jensen_shannontfp.vi.mutual_information.lower_bound_nguyen_wainwright_jordantfp.vi.pearsontfp.vi.squared_hellingertfp.vi.symmetrized_csiszar_functiontfp.vi.t_powertfp.vi.total_variationtfp.vi.triangular