LICENSE
MANIFEST.in
README.md
setup.py
distrax/__init__.py
distrax/distrax_test.py
distrax.egg-info/PKG-INFO
distrax.egg-info/SOURCES.txt
distrax.egg-info/dependency_links.txt
distrax.egg-info/not-zip-safe
distrax.egg-info/requires.txt
distrax.egg-info/top_level.txt
distrax/_src/__init__.py
distrax/_src/bijectors/__init__.py
distrax/_src/bijectors/bijector.py
distrax/_src/bijectors/bijector_from_tfp.py
distrax/_src/bijectors/bijector_from_tfp_test.py
distrax/_src/bijectors/bijector_test.py
distrax/_src/bijectors/block.py
distrax/_src/bijectors/block_test.py
distrax/_src/bijectors/chain.py
distrax/_src/bijectors/chain_test.py
distrax/_src/bijectors/diag_linear.py
distrax/_src/bijectors/diag_linear_test.py
distrax/_src/bijectors/diag_plus_low_rank_linear.py
distrax/_src/bijectors/diag_plus_low_rank_linear_test.py
distrax/_src/bijectors/gumbel_cdf.py
distrax/_src/bijectors/gumbel_cdf_test.py
distrax/_src/bijectors/inverse.py
distrax/_src/bijectors/inverse_test.py
distrax/_src/bijectors/lambda_bijector.py
distrax/_src/bijectors/lambda_bijector_test.py
distrax/_src/bijectors/lower_upper_triangular_affine.py
distrax/_src/bijectors/lower_upper_triangular_affine_test.py
distrax/_src/bijectors/masked_coupling.py
distrax/_src/bijectors/masked_coupling_test.py
distrax/_src/bijectors/rational_quadratic_spline.py
distrax/_src/bijectors/rational_quadratic_spline_float64_test.py
distrax/_src/bijectors/rational_quadratic_spline_test.py
distrax/_src/bijectors/scalar_affine.py
distrax/_src/bijectors/scalar_affine_test.py
distrax/_src/bijectors/shift.py
distrax/_src/bijectors/shift_test.py
distrax/_src/bijectors/sigmoid.py
distrax/_src/bijectors/sigmoid_test.py
distrax/_src/bijectors/split_coupling.py
distrax/_src/bijectors/split_coupling_test.py
distrax/_src/bijectors/tanh.py
distrax/_src/bijectors/tanh_test.py
distrax/_src/bijectors/tfp_compatible_bijector.py
distrax/_src/bijectors/tfp_compatible_bijector_test.py
distrax/_src/bijectors/triangular_linear.py
distrax/_src/bijectors/triangular_linear_test.py
distrax/_src/bijectors/unconstrained_affine.py
distrax/_src/bijectors/unconstrained_affine_test.py
distrax/_src/distributions/__init__.py
distrax/_src/distributions/bernoulli.py
distrax/_src/distributions/bernoulli_test.py
distrax/_src/distributions/categorical.py
distrax/_src/distributions/categorical_test.py
distrax/_src/distributions/deterministic.py
distrax/_src/distributions/deterministic_test.py
distrax/_src/distributions/distribution.py
distrax/_src/distributions/distribution_from_tfp.py
distrax/_src/distributions/distribution_from_tfp_test.py
distrax/_src/distributions/distribution_test.py
distrax/_src/distributions/epsilon_greedy.py
distrax/_src/distributions/epsilon_greedy_test.py
distrax/_src/distributions/gamma.py
distrax/_src/distributions/gamma_test.py
distrax/_src/distributions/greedy.py
distrax/_src/distributions/greedy_test.py
distrax/_src/distributions/gumbel.py
distrax/_src/distributions/gumbel_test.py
distrax/_src/distributions/independent.py
distrax/_src/distributions/independent_test.py
distrax/_src/distributions/laplace.py
distrax/_src/distributions/laplace_test.py
distrax/_src/distributions/log_stddev_normal.py
distrax/_src/distributions/log_stddev_normal_test.py
distrax/_src/distributions/logistic.py
distrax/_src/distributions/logistic_test.py
distrax/_src/distributions/mixture_same_family.py
distrax/_src/distributions/mixture_same_family_test.py
distrax/_src/distributions/multinomial.py
distrax/_src/distributions/multinomial_test.py
distrax/_src/distributions/mvn_diag.py
distrax/_src/distributions/mvn_diag_plus_low_rank.py
distrax/_src/distributions/mvn_diag_plus_low_rank_test.py
distrax/_src/distributions/mvn_diag_test.py
distrax/_src/distributions/mvn_from_bijector.py
distrax/_src/distributions/mvn_from_bijector_test.py
distrax/_src/distributions/mvn_full_covariance.py
distrax/_src/distributions/mvn_full_covariance_test.py
distrax/_src/distributions/mvn_kl_test.py
distrax/_src/distributions/mvn_tri.py
distrax/_src/distributions/mvn_tri_test.py
distrax/_src/distributions/normal.py
distrax/_src/distributions/normal_float64_test.py
distrax/_src/distributions/normal_test.py
distrax/_src/distributions/one_hot_categorical.py
distrax/_src/distributions/one_hot_categorical_test.py
distrax/_src/distributions/quantized.py
distrax/_src/distributions/quantized_test.py
distrax/_src/distributions/softmax.py
distrax/_src/distributions/softmax_test.py
distrax/_src/distributions/straight_through.py
distrax/_src/distributions/straight_through_test.py
distrax/_src/distributions/tfp_compatible_distribution.py
distrax/_src/distributions/tfp_compatible_distribution_test.py
distrax/_src/distributions/transformed.py
distrax/_src/distributions/transformed_test.py
distrax/_src/distributions/uniform.py
distrax/_src/distributions/uniform_float64_test.py
distrax/_src/distributions/uniform_test.py
distrax/_src/utils/__init__.py
distrax/_src/utils/conversion.py
distrax/_src/utils/conversion_test.py
distrax/_src/utils/equivalence.py
distrax/_src/utils/hmm.py
distrax/_src/utils/hmm_test.py
distrax/_src/utils/importance_sampling.py
distrax/_src/utils/importance_sampling_test.py
distrax/_src/utils/jittable.py
distrax/_src/utils/jittable_test.py
distrax/_src/utils/math.py
distrax/_src/utils/math_test.py
distrax/_src/utils/monte_carlo.py
distrax/_src/utils/monte_carlo_test.py
distrax/_src/utils/transformations.py
distrax/_src/utils/transformations_test.py
examples/flow.py
examples/hmm.py
examples/vae.py
requirements/requirements-examples.txt
requirements/requirements-tests.txt
requirements/requirements.txt