Path Lines of Code conftest.py 16 discussion/neutra/__init__.py 2 discussion/neutra/neutra_kernel.py 208 discussion/pathfinder/pathfinder.py 249 discussion/turnkey_inference_candidate/__init__.py 2 discussion/turnkey_inference_candidate/window_tune_nuts_sampling.py 305 required_packages.py 11 setup.py 74 spinoffs/fun_mc/fun_mc/__init__.py 1 spinoffs/fun_mc/fun_mc/api.py 8 spinoffs/fun_mc/fun_mc/backend.py 1 spinoffs/fun_mc/fun_mc/backends/__init__.py 1 spinoffs/fun_mc/fun_mc/backends/rewrite.py 109 spinoffs/fun_mc/fun_mc/dynamic/__init__.py 1 spinoffs/fun_mc/fun_mc/dynamic/backend_jax/__init__.py 1 spinoffs/fun_mc/fun_mc/dynamic/backend_jax/backend.py 13 spinoffs/fun_mc/fun_mc/dynamic/backend_jax/tf_on_jax.py 133 spinoffs/fun_mc/fun_mc/dynamic/backend_jax/util.py 159 spinoffs/fun_mc/fun_mc/dynamic/backend_tensorflow/__init__.py 1 spinoffs/fun_mc/fun_mc/dynamic/backend_tensorflow/backend.py 12 spinoffs/fun_mc/fun_mc/dynamic/backend_tensorflow/util.py 175 spinoffs/fun_mc/fun_mc/fun_mc_lib.py 1428 spinoffs/fun_mc/fun_mc/prefab.py 502 spinoffs/fun_mc/fun_mc/sga_hmc.py 340 spinoffs/fun_mc/fun_mc/using_jax.py 5 spinoffs/fun_mc/fun_mc/using_tensorflow.py 5 spinoffs/fun_mc/fun_mc/util.py 1 spinoffs/fun_mc/fun_mc/util_tfp.py 46 spinoffs/inference_gym/conftest.py 13 spinoffs/inference_gym/inference_gym/__init__.py 1 spinoffs/inference_gym/inference_gym/backends/__init__.py 1 spinoffs/inference_gym/inference_gym/backends/rewrite.py 151 spinoffs/inference_gym/inference_gym/backends/util.py 14 spinoffs/inference_gym/inference_gym/dynamic/__init__.py 1 spinoffs/inference_gym/inference_gym/dynamic/backend_jax/__init__.py 1 spinoffs/inference_gym/inference_gym/dynamic/backend_numpy/__init__.py 1 spinoffs/inference_gym/inference_gym/dynamic/backend_tensorflow/__init__.py 1 spinoffs/inference_gym/inference_gym/internal/__init__.py 1 spinoffs/inference_gym/inference_gym/internal/array_to_source.py 13 spinoffs/inference_gym/inference_gym/internal/data.py 320 spinoffs/inference_gym/inference_gym/internal/datasets/__init__.py 1 spinoffs/inference_gym/inference_gym/internal/datasets/brownian_motion_missing_middle_observations.py 11 spinoffs/inference_gym/inference_gym/internal/datasets/convection_lorenz_bridge.py 14 spinoffs/inference_gym/inference_gym/internal/datasets/sp500_closing_prices.py 2521 spinoffs/inference_gym/inference_gym/internal/datasets/synthetic_item_response_theory.py 90044 spinoffs/inference_gym/inference_gym/internal/datasets/synthetic_log_gaussian_cox_process.py 408 spinoffs/inference_gym/inference_gym/internal/datasets/synthetic_plasma_spectroscopy.py 1706 spinoffs/inference_gym/inference_gym/internal/datasets/synthetic_plasma_spectroscopy_with_bump.py 486 spinoffs/inference_gym/inference_gym/internal/ground_truth_encoding.py 63 spinoffs/inference_gym/inference_gym/targets/__init__.py 88 spinoffs/inference_gym/inference_gym/targets/banana.py 60 spinoffs/inference_gym/inference_gym/targets/bayesian_model.py 21 spinoffs/inference_gym/inference_gym/targets/brownian_motion.py 188 spinoffs/inference_gym/inference_gym/targets/eight_schools.py 56 spinoffs/inference_gym/inference_gym/targets/ground_truth/__init__.py 1 spinoffs/inference_gym/inference_gym/targets/ground_truth/brownian_motion_missing_middle_observations.py 98 spinoffs/inference_gym/inference_gym/targets/ground_truth/brownian_motion_unknown_scales_missing_middle_observations.py 116 spinoffs/inference_gym/inference_gym/targets/ground_truth/convection_lorenz_bridge.py 278 spinoffs/inference_gym/inference_gym/targets/ground_truth/convection_lorenz_bridge_unknown_scales.py 296 spinoffs/inference_gym/inference_gym/targets/ground_truth/eight_schools.py 50 spinoffs/inference_gym/inference_gym/targets/ground_truth/german_credit_numeric_logistic_regression.py 83 spinoffs/inference_gym/inference_gym/targets/ground_truth/german_credit_numeric_probit_regression.py 83 spinoffs/inference_gym/inference_gym/targets/ground_truth/german_credit_numeric_sparse_logistic_regression.py 173 spinoffs/inference_gym/inference_gym/targets/ground_truth/radon_contextual_effects_indiana.py 323 spinoffs/inference_gym/inference_gym/targets/ground_truth/radon_contextual_effects_indiana_halfnormal.py 323 spinoffs/inference_gym/inference_gym/targets/ground_truth/radon_contextual_effects_minnesota.py 305 spinoffs/inference_gym/inference_gym/targets/ground_truth/radon_contextual_effects_minnesota_halfnormal.py 305 spinoffs/inference_gym/inference_gym/targets/ground_truth/stochastic_volatility_log_sp500.py 7583 spinoffs/inference_gym/inference_gym/targets/ground_truth/stochastic_volatility_log_sp500_small.py 335 spinoffs/inference_gym/inference_gym/targets/ground_truth/stochastic_volatility_sp500.py 7583 spinoffs/inference_gym/inference_gym/targets/ground_truth/stochastic_volatility_sp500_small.py 335 spinoffs/inference_gym/inference_gym/targets/ground_truth/synthetic_item_response_theory.py 1523 spinoffs/inference_gym/inference_gym/targets/ground_truth/synthetic_log_gaussian_cox_process.py 326 spinoffs/inference_gym/inference_gym/targets/ill_conditioned_gaussian.py 57 spinoffs/inference_gym/inference_gym/targets/item_response_theory.py 192 spinoffs/inference_gym/inference_gym/targets/log_gaussian_cox_process.py 101 spinoffs/inference_gym/inference_gym/targets/logistic_regression.py 87 spinoffs/inference_gym/inference_gym/targets/lorenz_system.py 249 spinoffs/inference_gym/inference_gym/targets/model.py 100 spinoffs/inference_gym/inference_gym/targets/neals_funnel.py 58 spinoffs/inference_gym/inference_gym/targets/non_identifiable_quartic.py 49 spinoffs/inference_gym/inference_gym/targets/plasma_spectroscopy.py 223 spinoffs/inference_gym/inference_gym/targets/probit_regression.py 87 spinoffs/inference_gym/inference_gym/targets/radon_contextual_effects.py 242 spinoffs/inference_gym/inference_gym/targets/sparse_logistic_regression.py 113 spinoffs/inference_gym/inference_gym/targets/stochastic_volatility.py 149 spinoffs/inference_gym/inference_gym/targets/vector_model.py 126 spinoffs/inference_gym/inference_gym/targets/vectorized_stochastic_volatility.py 243 spinoffs/inference_gym/inference_gym/tools/__init__.py 1 spinoffs/inference_gym/inference_gym/tools/get_ground_truth.py 98 spinoffs/inference_gym/inference_gym/tools/stan/brownian_motion.py 51 spinoffs/inference_gym/inference_gym/tools/stan/eight_schools.py 30 spinoffs/inference_gym/inference_gym/tools/stan/item_response_theory.py 70 spinoffs/inference_gym/inference_gym/tools/stan/log_gaussian_cox_process.py 42 spinoffs/inference_gym/inference_gym/tools/stan/logistic_regression.py 46 spinoffs/inference_gym/inference_gym/tools/stan/lorenz_system.py 72 spinoffs/inference_gym/inference_gym/tools/stan/probit_regression.py 47 spinoffs/inference_gym/inference_gym/tools/stan/radon_contextual_effects.py 40 spinoffs/inference_gym/inference_gym/tools/stan/radon_contextual_effects_halfnormal.py 40 spinoffs/inference_gym/inference_gym/tools/stan/sparse_logistic_regression.py 60 spinoffs/inference_gym/inference_gym/tools/stan/stan_model.py 10 spinoffs/inference_gym/inference_gym/tools/stan/stochastic_volatility.py 29 spinoffs/inference_gym/inference_gym/tools/stan/targets.py 110 spinoffs/inference_gym/inference_gym/tools/stan/util.py 62 spinoffs/inference_gym/inference_gym/using_jax.py 6 spinoffs/inference_gym/inference_gym/using_numpy.py 6 spinoffs/inference_gym/inference_gym/using_tensorflow.py 6 spinoffs/inference_gym/inference_gym/version.py 11 spinoffs/inference_gym/setup.py 44 spinoffs/oryx/oryx/__init__.py 7 spinoffs/oryx/oryx/bijectors/__init__.py 12 spinoffs/oryx/oryx/bijectors/bijector_extensions.py 113 spinoffs/oryx/oryx/core/__init__.py 28 spinoffs/oryx/oryx/core/interpreters/__init__.py 5 spinoffs/oryx/oryx/core/interpreters/harvest.py 735 spinoffs/oryx/oryx/core/interpreters/inverse/__init__.py 4 spinoffs/oryx/oryx/core/interpreters/inverse/core.py 262 spinoffs/oryx/oryx/core/interpreters/inverse/custom_inverse.py 84 spinoffs/oryx/oryx/core/interpreters/inverse/rules.py 160 spinoffs/oryx/oryx/core/interpreters/inverse/slice.py 74 spinoffs/oryx/oryx/core/interpreters/log_prob.py 72 spinoffs/oryx/oryx/core/interpreters/propagate.py 212 spinoffs/oryx/oryx/core/interpreters/unzip.py 543 spinoffs/oryx/oryx/core/kwargs_util.py 41 spinoffs/oryx/oryx/core/ppl/__init__.py 17 spinoffs/oryx/oryx/core/ppl/effect_handler.py 96 spinoffs/oryx/oryx/core/ppl/plate_util.py 25 spinoffs/oryx/oryx/core/ppl/transformations.py 118 spinoffs/oryx/oryx/core/primitive.py 157 spinoffs/oryx/oryx/core/pytree.py 19 spinoffs/oryx/oryx/core/serialize.py 17 spinoffs/oryx/oryx/core/state/__init__.py 17 spinoffs/oryx/oryx/core/state/api.py 42 spinoffs/oryx/oryx/core/state/function.py 190 spinoffs/oryx/oryx/core/state/module.py 56 spinoffs/oryx/oryx/core/state/registrations.py 70 spinoffs/oryx/oryx/core/trace_util.py 77 spinoffs/oryx/oryx/distributions/__init__.py 8 spinoffs/oryx/oryx/distributions/distribution_extensions.py 85 spinoffs/oryx/oryx/experimental/__init__.py 4 spinoffs/oryx/oryx/experimental/autoconj/addn.py 47 spinoffs/oryx/oryx/experimental/autoconj/canonicalize.py 101 spinoffs/oryx/oryx/experimental/autoconj/einsum.py 94 spinoffs/oryx/oryx/experimental/matching/__init__.py 3 spinoffs/oryx/oryx/experimental/matching/jax_rewrite.py 348 spinoffs/oryx/oryx/experimental/matching/matcher.py 241 spinoffs/oryx/oryx/experimental/matching/rules.py 103 spinoffs/oryx/oryx/experimental/mcmc/__init__.py 7 spinoffs/oryx/oryx/experimental/mcmc/kernels.py 121 spinoffs/oryx/oryx/experimental/mcmc/utils.py 18 spinoffs/oryx/oryx/experimental/nn/__init__.py 21 spinoffs/oryx/oryx/experimental/nn/base.py 265 spinoffs/oryx/oryx/experimental/nn/combinator.py 52 spinoffs/oryx/oryx/experimental/nn/convolution.py 155 spinoffs/oryx/oryx/experimental/nn/core.py 101 spinoffs/oryx/oryx/experimental/nn/function.py 16 spinoffs/oryx/oryx/experimental/nn/normalization.py 103 spinoffs/oryx/oryx/experimental/nn/pooling.py 72 spinoffs/oryx/oryx/experimental/nn/reshape.py 35 spinoffs/oryx/oryx/experimental/optimizers/__init__.py 6 spinoffs/oryx/oryx/experimental/optimizers/optix.py 211 spinoffs/oryx/oryx/internal/__init__.py 1 spinoffs/oryx/oryx/tools/build_oryx_docs.py 28 spinoffs/oryx/oryx/util/__init__.py 2 spinoffs/oryx/oryx/util/summary.py 7 spinoffs/oryx/oryx/version.py 11 spinoffs/oryx/pyproject.toml 31 tensorflow_probability/__init__.py 3 tensorflow_probability/g3doc/_book.yaml 22 tensorflow_probability/g3doc/_index.yaml 101 tensorflow_probability/python/__init__.py 79 tensorflow_probability/python/bijectors/__init__.py 162 tensorflow_probability/python/bijectors/absolute_value.py 38 tensorflow_probability/python/bijectors/ascending.py 40 tensorflow_probability/python/bijectors/batch_normalization.py 106 tensorflow_probability/python/bijectors/bijector.py 914 tensorflow_probability/python/bijectors/bijector_test_util.py 175 tensorflow_probability/python/bijectors/blockwise.py 174 tensorflow_probability/python/bijectors/categorical_to_discrete.py 94 tensorflow_probability/python/bijectors/chain.py 65 tensorflow_probability/python/bijectors/cholesky_outer_product.py 83 tensorflow_probability/python/bijectors/cholesky_to_inv_cholesky.py 60 tensorflow_probability/python/bijectors/composition.py 468 tensorflow_probability/python/bijectors/correlation_cholesky.py 77 tensorflow_probability/python/bijectors/cumsum.py 51 tensorflow_probability/python/bijectors/discrete_cosine_transform.py 36 tensorflow_probability/python/bijectors/exp.py 27 tensorflow_probability/python/bijectors/expm1.py 42 tensorflow_probability/python/bijectors/ffjord.py 128 tensorflow_probability/python/bijectors/fill_scale_tril.py 40 tensorflow_probability/python/bijectors/fill_triangular.py 90 tensorflow_probability/python/bijectors/frechet_cdf.py 105 tensorflow_probability/python/bijectors/generalized_pareto.py 86 tensorflow_probability/python/bijectors/gev_cdf.py 134 tensorflow_probability/python/bijectors/glow.py 398 tensorflow_probability/python/bijectors/gompertz_cdf.py 89 tensorflow_probability/python/bijectors/gumbel_cdf.py 76 tensorflow_probability/python/bijectors/householder.py 41 tensorflow_probability/python/bijectors/hypothesis_testlib.py 402 tensorflow_probability/python/bijectors/identity.py 36 tensorflow_probability/python/bijectors/inline.py 65 tensorflow_probability/python/bijectors/invert.py 90 tensorflow_probability/python/bijectors/iterated_sigmoid_centered.py 62 tensorflow_probability/python/bijectors/joint_map.py 58 tensorflow_probability/python/bijectors/kumaraswamy_cdf.py 89 tensorflow_probability/python/bijectors/lambertw_transform.py 93 tensorflow_probability/python/bijectors/ldj_ratio.py 119 tensorflow_probability/python/bijectors/masked_autoregressive.py 496 tensorflow_probability/python/bijectors/matrix_inverse_tril.py 54 tensorflow_probability/python/bijectors/moyal_cdf.py 81 tensorflow_probability/python/bijectors/normal_cdf.py 44 tensorflow_probability/python/bijectors/ordered.py 45 tensorflow_probability/python/bijectors/pad.py 216 tensorflow_probability/python/bijectors/permute.py 91 tensorflow_probability/python/bijectors/power.py 63 tensorflow_probability/python/bijectors/power_transform.py 69 tensorflow_probability/python/bijectors/rational_quadratic_spline.py 247 tensorflow_probability/python/bijectors/rayleigh_cdf.py 80 tensorflow_probability/python/bijectors/real_nvp.py 174 tensorflow_probability/python/bijectors/reciprocal.py 34 tensorflow_probability/python/bijectors/reshape.py 206 tensorflow_probability/python/bijectors/restructure.py 123 tensorflow_probability/python/bijectors/scale.py 82 tensorflow_probability/python/bijectors/scale_matvec_diag.py 44 tensorflow_probability/python/bijectors/scale_matvec_linear_operator.py 121 tensorflow_probability/python/bijectors/scale_matvec_lu.py 105 tensorflow_probability/python/bijectors/scale_matvec_tril.py 40 tensorflow_probability/python/bijectors/shift.py 40 tensorflow_probability/python/bijectors/shifted_gompertz_cdf.py 99 tensorflow_probability/python/bijectors/sigmoid.py 96 tensorflow_probability/python/bijectors/sinh.py 27 tensorflow_probability/python/bijectors/sinh_arcsinh.py 83 tensorflow_probability/python/bijectors/soft_clip.py 125 tensorflow_probability/python/bijectors/softfloor.py 105 tensorflow_probability/python/bijectors/softmax_centered.py 61 tensorflow_probability/python/bijectors/softplus.py 108 tensorflow_probability/python/bijectors/softsign.py 47 tensorflow_probability/python/bijectors/split.py 308 tensorflow_probability/python/bijectors/square.py 36 tensorflow_probability/python/bijectors/tanh.py 27 tensorflow_probability/python/bijectors/transform_diagonal.py 39 tensorflow_probability/python/bijectors/transpose.py 229 tensorflow_probability/python/bijectors/weibull_cdf.py 97 tensorflow_probability/python/build_defs.bzl 363 tensorflow_probability/python/debugging/__init__.py 4 tensorflow_probability/python/debugging/benchmarking/__init__.py 20 tensorflow_probability/python/debugging/benchmarking/benchmark_tf_function.py 92 tensorflow_probability/python/distributions/__init__.py 263 tensorflow_probability/python/distributions/autoregressive.py 150 tensorflow_probability/python/distributions/batch_broadcast.py 233 tensorflow_probability/python/distributions/batch_concat.py 293 tensorflow_probability/python/distributions/batch_reshape.py 414 tensorflow_probability/python/distributions/bates.py 237 tensorflow_probability/python/distributions/bernoulli.py 174 tensorflow_probability/python/distributions/beta.py 224 tensorflow_probability/python/distributions/beta_binomial.py 140 tensorflow_probability/python/distributions/beta_quotient.py 160 tensorflow_probability/python/distributions/binomial.py 318 tensorflow_probability/python/distributions/blockwise.py 248 tensorflow_probability/python/distributions/categorical.py 242 tensorflow_probability/python/distributions/cauchy.py 122 tensorflow_probability/python/distributions/chi.py 81 tensorflow_probability/python/distributions/chi2.py 139 tensorflow_probability/python/distributions/cholesky_lkj.py 133 tensorflow_probability/python/distributions/cholesky_util.py 30 tensorflow_probability/python/distributions/continuous_bernoulli.py 253 tensorflow_probability/python/distributions/deterministic.py 210 tensorflow_probability/python/distributions/dirichlet.py 185 tensorflow_probability/python/distributions/dirichlet_multinomial.py 166 tensorflow_probability/python/distributions/discrete_rejection_sampling.py 64 tensorflow_probability/python/distributions/distribution.py 879 tensorflow_probability/python/distributions/doublesided_maxwell.py 104 tensorflow_probability/python/distributions/dpp.py 204 tensorflow_probability/python/distributions/empirical.py 186 tensorflow_probability/python/distributions/exp_gamma.py 186 tensorflow_probability/python/distributions/exponential.py 70 tensorflow_probability/python/distributions/exponentially_modified_gaussian.py 131 tensorflow_probability/python/distributions/finite_discrete.py 226 tensorflow_probability/python/distributions/gamma.py 479 tensorflow_probability/python/distributions/gamma_gamma.py 165 tensorflow_probability/python/distributions/gaussian_process.py 326 tensorflow_probability/python/distributions/gaussian_process_regression_model.py 267 tensorflow_probability/python/distributions/generalized_normal.py 169 tensorflow_probability/python/distributions/generalized_pareto.py 173 tensorflow_probability/python/distributions/geometric.py 164 tensorflow_probability/python/distributions/gev.py 121 tensorflow_probability/python/distributions/gumbel.py 83 tensorflow_probability/python/distributions/half_cauchy.py 127 tensorflow_probability/python/distributions/half_normal.py 106 tensorflow_probability/python/distributions/half_student_t.py 179 tensorflow_probability/python/distributions/hidden_markov_model.py 717 tensorflow_probability/python/distributions/horseshoe.py 101 tensorflow_probability/python/distributions/hypothesis_testlib.py 910 tensorflow_probability/python/distributions/independent.py 253 tensorflow_probability/python/distributions/internal/__init__.py 1 tensorflow_probability/python/distributions/internal/correlation_matrix_volumes.py 32 tensorflow_probability/python/distributions/internal/correlation_matrix_volumes_lib.py 75 tensorflow_probability/python/distributions/internal/statistical_testing.py 471 tensorflow_probability/python/distributions/inverse_gamma.py 151 tensorflow_probability/python/distributions/inverse_gaussian.py 197 tensorflow_probability/python/distributions/johnson_su.py 111 tensorflow_probability/python/distributions/joint_distribution.py 921 tensorflow_probability/python/distributions/joint_distribution_auto_batched.py 49 tensorflow_probability/python/distributions/joint_distribution_coroutine.py 47 tensorflow_probability/python/distributions/joint_distribution_named.py 119 tensorflow_probability/python/distributions/joint_distribution_sequential.py 235 tensorflow_probability/python/distributions/joint_distribution_util.py 49 tensorflow_probability/python/distributions/kullback_leibler.py 131 tensorflow_probability/python/distributions/kumaraswamy.py 136 tensorflow_probability/python/distributions/lambertw_f.py 163 tensorflow_probability/python/distributions/laplace.py 132 tensorflow_probability/python/distributions/linear_gaussian_ssm.py 1010 tensorflow_probability/python/distributions/lkj.py 297 tensorflow_probability/python/distributions/log_prob_ratio.py 21 tensorflow_probability/python/distributions/logistic.py 106 tensorflow_probability/python/distributions/logitnormal.py 230 tensorflow_probability/python/distributions/loglogistic.py 105 tensorflow_probability/python/distributions/lognormal.py 80 tensorflow_probability/python/distributions/markov_chain.py 368 tensorflow_probability/python/distributions/masked.py 226 tensorflow_probability/python/distributions/matrix_normal_linear_operator.py 124 tensorflow_probability/python/distributions/matrix_t_linear_operator.py 133 tensorflow_probability/python/distributions/mixture.py 242 tensorflow_probability/python/distributions/mixture_same_family.py 333 tensorflow_probability/python/distributions/moyal.py 90 tensorflow_probability/python/distributions/multinomial.py 178 tensorflow_probability/python/distributions/multivariate_student_t.py 197 tensorflow_probability/python/distributions/mvn_diag.py 102 tensorflow_probability/python/distributions/mvn_diag_plus_low_rank.py 89 tensorflow_probability/python/distributions/mvn_diag_plus_low_rank_covariance.py 73 tensorflow_probability/python/distributions/mvn_full_covariance.py 73 tensorflow_probability/python/distributions/mvn_linear_operator.py 160 tensorflow_probability/python/distributions/mvn_low_rank_update_linear_operator_covariance.py 187 tensorflow_probability/python/distributions/mvn_tril.py 78 tensorflow_probability/python/distributions/negative_binomial.py 174 tensorflow_probability/python/distributions/normal.py 125 tensorflow_probability/python/distributions/normal_conjugate_posteriors.py 68 tensorflow_probability/python/distributions/normal_inverse_gaussian.py 161 tensorflow_probability/python/distributions/onehot_categorical.py 207 tensorflow_probability/python/distributions/ordered_logistic.py 144 tensorflow_probability/python/distributions/pareto.py 174 tensorflow_probability/python/distributions/pert.py 149 tensorflow_probability/python/distributions/pixel_cnn.py 425 tensorflow_probability/python/distributions/plackett_luce.py 141 tensorflow_probability/python/distributions/poisson.py 291 tensorflow_probability/python/distributions/poisson_lognormal.py 215 tensorflow_probability/python/distributions/power_spherical.py 258 tensorflow_probability/python/distributions/probit_bernoulli.py 139 tensorflow_probability/python/distributions/quantized_distribution.py 232 tensorflow_probability/python/distributions/relaxed_bernoulli.py 161 tensorflow_probability/python/distributions/relaxed_onehot_categorical.py 262 tensorflow_probability/python/distributions/sample.py 370 tensorflow_probability/python/distributions/sigmoid_beta.py 116 tensorflow_probability/python/distributions/sinh_arcsinh.py 84 tensorflow_probability/python/distributions/skellam.py 187 tensorflow_probability/python/distributions/spherical_uniform.py 127 tensorflow_probability/python/distributions/stopping_ratio_logistic.py 140 tensorflow_probability/python/distributions/student_t.py 178 tensorflow_probability/python/distributions/student_t_process.py 286 tensorflow_probability/python/distributions/student_t_process_regression_model.py 344 tensorflow_probability/python/distributions/transformed_distribution.py 320 tensorflow_probability/python/distributions/triangular.py 143 tensorflow_probability/python/distributions/truncated_cauchy.py 220 tensorflow_probability/python/distributions/truncated_normal.py 290 tensorflow_probability/python/distributions/uniform.py 143 tensorflow_probability/python/distributions/untestable_distributions.py 1 tensorflow_probability/python/distributions/variational_gaussian_process.py 419 tensorflow_probability/python/distributions/vector_exponential_linear_operator.py 141 tensorflow_probability/python/distributions/von_mises.py 301 tensorflow_probability/python/distributions/von_mises_fisher.py 333 tensorflow_probability/python/distributions/weibull.py 131 tensorflow_probability/python/distributions/wishart.py 344 tensorflow_probability/python/distributions/zipf.py 201 tensorflow_probability/python/experimental/__init__.py 47 tensorflow_probability/python/experimental/auto_batching/__init__.py 42 tensorflow_probability/python/experimental/auto_batching/allocation_strategy.py 93 tensorflow_probability/python/experimental/auto_batching/backend_test_lib.py 93 tensorflow_probability/python/experimental/auto_batching/dsl.py 224 tensorflow_probability/python/experimental/auto_batching/frontend.py 373 tensorflow_probability/python/experimental/auto_batching/gast_util.py 33 tensorflow_probability/python/experimental/auto_batching/instructions.py 599 tensorflow_probability/python/experimental/auto_batching/liveness.py 48 tensorflow_probability/python/experimental/auto_batching/lowering.py 178 tensorflow_probability/python/experimental/auto_batching/numpy_backend.py 159 tensorflow_probability/python/experimental/auto_batching/stack_optimization.py 34 tensorflow_probability/python/experimental/auto_batching/stackless.py 144 tensorflow_probability/python/experimental/auto_batching/tf_backend.py 333 tensorflow_probability/python/experimental/auto_batching/type_inference.py 166 tensorflow_probability/python/experimental/auto_batching/virtual_machine.py 155 tensorflow_probability/python/experimental/auto_batching/xla.py 16 tensorflow_probability/python/experimental/bijectors/__init__.py 16 tensorflow_probability/python/experimental/bijectors/distribution_bijectors.py 85 tensorflow_probability/python/experimental/bijectors/highway_flow.py 210 tensorflow_probability/python/experimental/bijectors/scalar_function_with_inferred_inverse.py 204 tensorflow_probability/python/experimental/bijectors/sharded.py 33 tensorflow_probability/python/experimental/distribute/__init__.py 17 tensorflow_probability/python/experimental/distribute/joint_distribution.py 107 tensorflow_probability/python/experimental/distribute/sharded.py 109 tensorflow_probability/python/experimental/distributions/__init__.py 13 tensorflow_probability/python/experimental/distributions/importance_resample.py 252 tensorflow_probability/python/experimental/distributions/increment_log_prob.py 113 tensorflow_probability/python/experimental/distributions/joint_distribution_pinned.py 257 tensorflow_probability/python/experimental/distributions/marginal_fns.py 91 tensorflow_probability/python/experimental/distributions/mvn_precision_factor_linop.py 119 tensorflow_probability/python/experimental/joint_distribution_layers/__init__.py 3 tensorflow_probability/python/experimental/joint_distribution_layers/layers.py 154 tensorflow_probability/python/experimental/linalg/__init__.py 10 tensorflow_probability/python/experimental/linalg/linear_operator_psd_kernel.py 303 tensorflow_probability/python/experimental/linalg/no_pivot_ldl.py 65 tensorflow_probability/python/experimental/marginalize/__init__.py 6 tensorflow_probability/python/experimental/marginalize/logeinsumexp.py 82 tensorflow_probability/python/experimental/marginalize/marginalizable.py 117 tensorflow_probability/python/experimental/math/__init__.py 16 tensorflow_probability/python/experimental/math/manual_special_functions.py 222 tensorflow_probability/python/experimental/mcmc/__init__.py 120 tensorflow_probability/python/experimental/mcmc/covariance_reducer.py 107 tensorflow_probability/python/experimental/mcmc/diagonal_mass_matrix_adaptation.py 177 tensorflow_probability/python/experimental/mcmc/elliptical_slice_sampler.py 212 tensorflow_probability/python/experimental/mcmc/expectations_reducer.py 80 tensorflow_probability/python/experimental/mcmc/gradient_based_trajectory_length_adaptation.py 536 tensorflow_probability/python/experimental/mcmc/initialization.py 116 tensorflow_probability/python/experimental/mcmc/internal/__init__.py 1 tensorflow_probability/python/experimental/mcmc/kernel_builder.py 317 tensorflow_probability/python/experimental/mcmc/kernel_outputs.py 48 tensorflow_probability/python/experimental/mcmc/nuts_autobatching.py 388 tensorflow_probability/python/experimental/mcmc/particle_filter.py 256 tensorflow_probability/python/experimental/mcmc/particle_filter_augmentation.py 74 tensorflow_probability/python/experimental/mcmc/potential_scale_reduction_reducer.py 69 tensorflow_probability/python/experimental/mcmc/preconditioned_hmc.py 199 tensorflow_probability/python/experimental/mcmc/preconditioned_nuts.py 826 tensorflow_probability/python/experimental/mcmc/preconditioning_utils.py 136 tensorflow_probability/python/experimental/mcmc/progress_bar_reducer.py 47 tensorflow_probability/python/experimental/mcmc/reducer.py 14 tensorflow_probability/python/experimental/mcmc/run.py 94 tensorflow_probability/python/experimental/mcmc/sample.py 94 tensorflow_probability/python/experimental/mcmc/sample_discarding_kernel.py 81 tensorflow_probability/python/experimental/mcmc/sample_fold.py 124 tensorflow_probability/python/experimental/mcmc/sample_sequential_monte_carlo.py 376 tensorflow_probability/python/experimental/mcmc/sequential_monte_carlo_kernel.py 129 tensorflow_probability/python/experimental/mcmc/sharded.py 41 tensorflow_probability/python/experimental/mcmc/snaper_hmc.py 784 tensorflow_probability/python/experimental/mcmc/step.py 46 tensorflow_probability/python/experimental/mcmc/thermodynamic_integrals.py 41 tensorflow_probability/python/experimental/mcmc/thinning_kernel.py 46 tensorflow_probability/python/experimental/mcmc/tracing_reducer.py 82 tensorflow_probability/python/experimental/mcmc/weighted_resampling.py 195 tensorflow_probability/python/experimental/mcmc/windowed_sampling.py 502 tensorflow_probability/python/experimental/mcmc/with_reductions.py 74 tensorflow_probability/python/experimental/nn/__init__.py 42 tensorflow_probability/python/experimental/nn/affine_layers.py 192 tensorflow_probability/python/experimental/nn/convolutional_layers.py 295 tensorflow_probability/python/experimental/nn/convolutional_layers_v2.py 168 tensorflow_probability/python/experimental/nn/convolutional_transpose_layers.py 184 tensorflow_probability/python/experimental/nn/initializers/__init__.py 12 tensorflow_probability/python/experimental/nn/initializers/initializers.py 77 tensorflow_probability/python/experimental/nn/layers.py 122 tensorflow_probability/python/experimental/nn/losses/__init__.py 12 tensorflow_probability/python/experimental/nn/losses/losses.py 35 tensorflow_probability/python/experimental/nn/util/__init__.py 54 tensorflow_probability/python/experimental/nn/util/convolution_util.py 707 tensorflow_probability/python/experimental/nn/util/kernel_bias.py 107 tensorflow_probability/python/experimental/nn/util/random_variable.py 65 tensorflow_probability/python/experimental/nn/util/utils.py 251 tensorflow_probability/python/experimental/nn/variational_base.py 165 tensorflow_probability/python/experimental/parallel_filter/__init__.py 6 tensorflow_probability/python/experimental/parallel_filter/parallel_kalman_filter_lib.py 411 tensorflow_probability/python/experimental/sequential/__init__.py 22 tensorflow_probability/python/experimental/sequential/ensemble_adjustment_kalman_filter.py 80 tensorflow_probability/python/experimental/sequential/ensemble_kalman_filter.py 118 tensorflow_probability/python/experimental/sequential/extended_kalman_filter.py 148 tensorflow_probability/python/experimental/sequential/iterated_filter.py 393 tensorflow_probability/python/experimental/stats/__init__.py 12 tensorflow_probability/python/experimental/stats/sample_stats.py 343 tensorflow_probability/python/experimental/sts_gibbs/__init__.py 1 tensorflow_probability/python/experimental/sts_gibbs/gibbs_sampler.py 441 tensorflow_probability/python/experimental/sts_gibbs/spike_and_slab.py 276 tensorflow_probability/python/experimental/substrates/__init__.py 13 tensorflow_probability/python/experimental/tangent_spaces/__init__.py 14 tensorflow_probability/python/experimental/tangent_spaces/spaces.py 70 tensorflow_probability/python/experimental/util/__init__.py 18 tensorflow_probability/python/experimental/util/composite_tensor.py 163 tensorflow_probability/python/experimental/util/deferred_module.py 25 tensorflow_probability/python/experimental/util/jit_public_methods.py 77 tensorflow_probability/python/experimental/util/special_methods.py 268 tensorflow_probability/python/experimental/util/trainable.py 73 tensorflow_probability/python/experimental/vi/__init__.py 28 tensorflow_probability/python/experimental/vi/automatic_structured_vi.py 305 tensorflow_probability/python/experimental/vi/surrogate_posteriors.py 273 tensorflow_probability/python/experimental/vi/util/__init__.py 14 tensorflow_probability/python/experimental/vi/util/trainable_linear_operators.py 179 tensorflow_probability/python/glm/__init__.py 46 tensorflow_probability/python/glm/family.py 276 tensorflow_probability/python/glm/fisher_scoring.py 280 tensorflow_probability/python/glm/proximal_hessian.py 117 tensorflow_probability/python/internal/__init__.py 1 tensorflow_probability/python/internal/all_util.py 13 tensorflow_probability/python/internal/assert_util.py 51 tensorflow_probability/python/internal/auto_composite_tensor.py 302 tensorflow_probability/python/internal/backend/__init__.py 1 tensorflow_probability/python/internal/backend/jax/rewrite.py 39 tensorflow_probability/python/internal/backend/meta/gen_linear_operators.py 182 tensorflow_probability/python/internal/backend/meta/gen_tensor_shape.py 44 tensorflow_probability/python/internal/backend/numpy/__init__.py 39 tensorflow_probability/python/internal/backend/numpy/__internal__.py 3 tensorflow_probability/python/internal/backend/numpy/_utils.py 82 tensorflow_probability/python/internal/backend/numpy/bitwise.py 12 tensorflow_probability/python/internal/backend/numpy/compat.py 10 tensorflow_probability/python/internal/backend/numpy/composite_tensor.py 3 tensorflow_probability/python/internal/backend/numpy/config.py 13 tensorflow_probability/python/internal/backend/numpy/control_flow.py 139 tensorflow_probability/python/internal/backend/numpy/data_structures.py 3 tensorflow_probability/python/internal/backend/numpy/debugging.py 227 tensorflow_probability/python/internal/backend/numpy/deprecation.py 23 tensorflow_probability/python/internal/backend/numpy/dtype.py 59 tensorflow_probability/python/internal/backend/numpy/errors.py 8 tensorflow_probability/python/internal/backend/numpy/functional_ops.py 124 tensorflow_probability/python/internal/backend/numpy/gen/__init__.py 1 tensorflow_probability/python/internal/backend/numpy/gen/adjoint_registrations.py 101 tensorflow_probability/python/internal/backend/numpy/gen/cholesky_registrations.py 78 tensorflow_probability/python/internal/backend/numpy/gen/inverse_registrations.py 141 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator.py 599 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_addition.py 172 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_adjoint.py 120 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_algebra.py 139 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_block_diag.py 369 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_block_lower_triangular.py 396 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_circulant.py 363 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_composition.py 136 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_diag.py 134 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_full_matrix.py 81 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_householder.py 137 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_identity.py 372 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_inversion.py 105 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_kronecker.py 301 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_low_rank_update.py 261 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_lower_triangular.py 93 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_toeplitz.py 142 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_util.py 247 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_zeros.py 260 tensorflow_probability/python/internal/backend/numpy/gen/matmul_registrations.py 183 tensorflow_probability/python/internal/backend/numpy/gen/registrations_util.py 47 tensorflow_probability/python/internal/backend/numpy/gen/solve_registrations.py 163 tensorflow_probability/python/internal/backend/numpy/gen/tensor_shape.py 464 tensorflow_probability/python/internal/backend/numpy/initializers.py 14 tensorflow_probability/python/internal/backend/numpy/keras_layers.py 5 tensorflow_probability/python/internal/backend/numpy/linalg.py 80 tensorflow_probability/python/internal/backend/numpy/linalg_impl.py 349 tensorflow_probability/python/internal/backend/numpy/misc.py 132 tensorflow_probability/python/internal/backend/numpy/nest.py 199 tensorflow_probability/python/internal/backend/numpy/nested_structure_coder.py 3 tensorflow_probability/python/internal/backend/numpy/nn.py 163 tensorflow_probability/python/internal/backend/numpy/numpy_array.py 370 tensorflow_probability/python/internal/backend/numpy/numpy_keras.py 4 tensorflow_probability/python/internal/backend/numpy/numpy_logging.py 103 tensorflow_probability/python/internal/backend/numpy/numpy_math.py 717 tensorflow_probability/python/internal/backend/numpy/numpy_signal.py 73 tensorflow_probability/python/internal/backend/numpy/ops.py 480 tensorflow_probability/python/internal/backend/numpy/private.py 76 tensorflow_probability/python/internal/backend/numpy/random_generators.py 228 tensorflow_probability/python/internal/backend/numpy/raw_ops.py 63 tensorflow_probability/python/internal/backend/numpy/resource_variable_ops.py 7 tensorflow_probability/python/internal/backend/numpy/sets_lib.py 16 tensorflow_probability/python/internal/backend/numpy/sparse_lib.py 15 tensorflow_probability/python/internal/backend/numpy/tensor_array_ops.py 129 tensorflow_probability/python/internal/backend/numpy/tensor_spec.py 9 tensorflow_probability/python/internal/backend/numpy/tf_inspect.py 26 tensorflow_probability/python/internal/backend/numpy/type_spec.py 20 tensorflow_probability/python/internal/backend/numpy/v1.py 119 tensorflow_probability/python/internal/backend/numpy/v2.py 93 tensorflow_probability/python/internal/backend/numpy/variables.py 3 tensorflow_probability/python/internal/batch_shape_lib.py 163 tensorflow_probability/python/internal/batched_rejection_sampler.py 53 tensorflow_probability/python/internal/broadcast_util.py 47 tensorflow_probability/python/internal/cache_util.py 242 tensorflow_probability/python/internal/callable_util.py 22 tensorflow_probability/python/internal/custom_gradient.py 72 tensorflow_probability/python/internal/distribute_lib.py 187 tensorflow_probability/python/internal/distribute_test_lib.py 71 tensorflow_probability/python/internal/distribution_util.py 684 tensorflow_probability/python/internal/docstring_util.py 15 tensorflow_probability/python/internal/dtype_util.py 173 tensorflow_probability/python/internal/hypothesis_testlib.py 538 tensorflow_probability/python/internal/implementation_selection.py 64 tensorflow_probability/python/internal/lazy_loader.py 27 tensorflow_probability/python/internal/loop_util.py 112 tensorflow_probability/python/internal/monte_carlo.py 45 tensorflow_probability/python/internal/name_util.py 65 tensorflow_probability/python/internal/nest_util.py 213 tensorflow_probability/python/internal/numerics_testing.py 109 tensorflow_probability/python/internal/parameter_properties.py 105 tensorflow_probability/python/internal/prefer_static.py 329 tensorflow_probability/python/internal/reparameterization.py 16 tensorflow_probability/python/internal/samplers.py 156 tensorflow_probability/python/internal/slicing.py 104 tensorflow_probability/python/internal/special_math.py 84 tensorflow_probability/python/internal/structural_tuple.py 100 tensorflow_probability/python/internal/tensor_util.py 55 tensorflow_probability/python/internal/tensorshape_util.py 76 tensorflow_probability/python/internal/trainable_state_util.py 138 tensorflow_probability/python/internal/unnest.py 117 tensorflow_probability/python/internal/variadic_reduce.py 120 tensorflow_probability/python/internal/vectorization_util.py 145 tensorflow_probability/python/layers/__init__.py 61 tensorflow_probability/python/layers/conv_variational.py 652 tensorflow_probability/python/layers/dense_variational.py 330 tensorflow_probability/python/layers/dense_variational_v2.py 91 tensorflow_probability/python/layers/distribution_layer.py 806 tensorflow_probability/python/layers/initializers.py 73 tensorflow_probability/python/layers/internal/__init__.py 1 tensorflow_probability/python/layers/internal/distribution_tensor_coercible.py 126 tensorflow_probability/python/layers/internal/tensor_tuple.py 53 tensorflow_probability/python/layers/masked_autoregressive.py 26 tensorflow_probability/python/layers/util.py 96 tensorflow_probability/python/layers/variable_input.py 37 tensorflow_probability/python/layers/weight_norm.py 82 tensorflow_probability/python/math/__init__.py 145 tensorflow_probability/python/math/bessel.py 692 tensorflow_probability/python/math/custom_gradient.py 40 tensorflow_probability/python/math/diag_jacobian.py 97 tensorflow_probability/python/math/generic.py 250 tensorflow_probability/python/math/gradient.py 240 tensorflow_probability/python/math/gram_schmidt.py 23 tensorflow_probability/python/math/hypergeometric.py 281 tensorflow_probability/python/math/integration.py 49 tensorflow_probability/python/math/interpolation.py 392 tensorflow_probability/python/math/linalg.py 468 tensorflow_probability/python/math/minimize.py 214 tensorflow_probability/python/math/numeric.py 23 tensorflow_probability/python/math/ode/__init__.py 14 tensorflow_probability/python/math/ode/base.py 289 tensorflow_probability/python/math/ode/bdf.py 582 tensorflow_probability/python/math/ode/bdf_util.py 188 tensorflow_probability/python/math/ode/dormand_prince.py 433 tensorflow_probability/python/math/ode/runge_kutta_util.py 117 tensorflow_probability/python/math/ode/util.py 110 tensorflow_probability/python/math/psd_kernels/__init__.py 44 tensorflow_probability/python/math/psd_kernels/changepoint.py 156 tensorflow_probability/python/math/psd_kernels/exp_sin_squared.py 86 tensorflow_probability/python/math/psd_kernels/exponential_curve.py 64 tensorflow_probability/python/math/psd_kernels/exponentiated_quadratic.py 106 tensorflow_probability/python/math/psd_kernels/feature_scaled.py 74 tensorflow_probability/python/math/psd_kernels/feature_transformed.py 45 tensorflow_probability/python/math/psd_kernels/hypothesis_testlib.py 561 tensorflow_probability/python/math/psd_kernels/internal/__init__.py 1 tensorflow_probability/python/math/psd_kernels/internal/util.py 99 tensorflow_probability/python/math/psd_kernels/kumaraswamy_transformed.py 85 tensorflow_probability/python/math/psd_kernels/matern.py 276 tensorflow_probability/python/math/psd_kernels/parabolic.py 85 tensorflow_probability/python/math/psd_kernels/pointwise_exponential.py 28 tensorflow_probability/python/math/psd_kernels/polynomial.py 183 tensorflow_probability/python/math/psd_kernels/positive_semidefinite_kernel.py 378 tensorflow_probability/python/math/psd_kernels/rational_quadratic.py 127 tensorflow_probability/python/math/psd_kernels/schur_complement.py 227 tensorflow_probability/python/math/root_search.py 251 tensorflow_probability/python/math/scan_associative.py 150 tensorflow_probability/python/math/sparse.py 19 tensorflow_probability/python/math/special.py 1004 tensorflow_probability/python/mcmc/__init__.py 54 tensorflow_probability/python/mcmc/diagnostic.py 200 tensorflow_probability/python/mcmc/dual_averaging_step_size_adaptation.py 301 tensorflow_probability/python/mcmc/eight_schools_hmc.py 106 tensorflow_probability/python/mcmc/hmc.py 360 tensorflow_probability/python/mcmc/internal/__init__.py 1 tensorflow_probability/python/mcmc/internal/leapfrog_integrator.py 155 tensorflow_probability/python/mcmc/internal/slice_sampler_utils.py 151 tensorflow_probability/python/mcmc/internal/util.py 286 tensorflow_probability/python/mcmc/kernel.py 25 tensorflow_probability/python/mcmc/langevin.py 451 tensorflow_probability/python/mcmc/metropolis_hastings.py 131 tensorflow_probability/python/mcmc/nuts.py 820 tensorflow_probability/python/mcmc/random_walk_metropolis.py 207 tensorflow_probability/python/mcmc/replica_exchange_mc.py 486 tensorflow_probability/python/mcmc/sample.py 101 tensorflow_probability/python/mcmc/sample_annealed_importance.py 112 tensorflow_probability/python/mcmc/sample_halton_sequence.py 105 tensorflow_probability/python/mcmc/simple_step_size_adaptation.py 193 tensorflow_probability/python/mcmc/slice_sampler_kernel.py 259 tensorflow_probability/python/mcmc/transformed_kernel.py 187 tensorflow_probability/python/monte_carlo/__init__.py 6 tensorflow_probability/python/monte_carlo/expectation.py 25 tensorflow_probability/python/optimizer/__init__.py 32 tensorflow_probability/python/optimizer/bfgs.py 161 tensorflow_probability/python/optimizer/bfgs_utils.py 159 tensorflow_probability/python/optimizer/convergence_criteria/__init__.py 10 tensorflow_probability/python/optimizer/convergence_criteria/convergence_criterion.py 34 tensorflow_probability/python/optimizer/convergence_criteria/loss_not_decreasing.py 63 tensorflow_probability/python/optimizer/convergence_criteria/successive_gradients_are_uncorrelated.py 25 tensorflow_probability/python/optimizer/differential_evolution.py 318 tensorflow_probability/python/optimizer/lbfgs.py 143 tensorflow_probability/python/optimizer/linesearch/__init__.py 6 tensorflow_probability/python/optimizer/linesearch/hager_zhang.py 219 tensorflow_probability/python/optimizer/linesearch/internal/__init__.py 1 tensorflow_probability/python/optimizer/linesearch/internal/hager_zhang_lib.py 285 tensorflow_probability/python/optimizer/nelder_mead.py 450 tensorflow_probability/python/optimizer/proximal_hessian_sparse.py 196 tensorflow_probability/python/optimizer/sgld.py 124 tensorflow_probability/python/optimizer/variational_sgd.py 153 tensorflow_probability/python/random/__init__.py 14 tensorflow_probability/python/random/random_ops.py 76 tensorflow_probability/python/stats/__init__.py 64 tensorflow_probability/python/stats/calibration.py 203 tensorflow_probability/python/stats/kendalls_tau.py 137 tensorflow_probability/python/stats/leave_one_out.py 82 tensorflow_probability/python/stats/moving_stats.py 99 tensorflow_probability/python/stats/quantiles.py 359 tensorflow_probability/python/stats/ranking.py 129 tensorflow_probability/python/stats/sample_stats.py 385 tensorflow_probability/python/sts/__init__.py 55 tensorflow_probability/python/sts/anomaly_detection/__init__.py 10 tensorflow_probability/python/sts/anomaly_detection/anomaly_detection_lib.py 214 tensorflow_probability/python/sts/components/__init__.py 50 tensorflow_probability/python/sts/components/autoregressive.py 155 tensorflow_probability/python/sts/components/autoregressive_integrated_moving_average.py 228 tensorflow_probability/python/sts/components/autoregressive_moving_average.py 96 tensorflow_probability/python/sts/components/dynamic_regression.py 122 tensorflow_probability/python/sts/components/local_level.py 140 tensorflow_probability/python/sts/components/local_linear_trend.py 183 tensorflow_probability/python/sts/components/regression.py 183 tensorflow_probability/python/sts/components/seasonal.py 330 tensorflow_probability/python/sts/components/semilocal_linear_trend.py 187 tensorflow_probability/python/sts/components/smooth_seasonal.py 170 tensorflow_probability/python/sts/components/sum.py 212 tensorflow_probability/python/sts/decomposition.py 77 tensorflow_probability/python/sts/default_model.py 40 tensorflow_probability/python/sts/fitting.py 129 tensorflow_probability/python/sts/forecast.py 141 tensorflow_probability/python/sts/holiday_effects.py 61 tensorflow_probability/python/sts/internal/__init__.py 1 tensorflow_probability/python/sts/internal/missing_values_util.py 52 tensorflow_probability/python/sts/internal/seasonality_util.py 123 tensorflow_probability/python/sts/internal/util.py 122 tensorflow_probability/python/sts/regularization.py 120 tensorflow_probability/python/sts/structural_time_series.py 219 tensorflow_probability/python/util/__init__.py 12 tensorflow_probability/python/util/deferred_tensor.py 515 tensorflow_probability/python/util/seed_stream.py 39 tensorflow_probability/python/version.py 11 tensorflow_probability/python/vi/__init__.py 50 tensorflow_probability/python/vi/csiszar_divergence.py 282 tensorflow_probability/python/vi/mutual_information.py 151 tensorflow_probability/python/vi/optimization.py 95 tensorflow_probability/substrates/__init__.py 13 tensorflow_probability/substrates/jax/__init__.py 25 tensorflow_probability/substrates/meta/rewrite.py 236 tensorflow_probability/substrates/numpy/__init__.py 15 tensorflow_probability/tools/build_docs.py 46 testing/virtualenv_is_active.py 6