README.md
pyproject.toml
src/tensorcontainer/__init__.py
src/tensorcontainer/tensor_annotated.py
src/tensorcontainer/tensor_container.py
src/tensorcontainer/tensor_dataclass.py
src/tensorcontainer/tensor_dict.py
src/tensorcontainer/types.py
src/tensorcontainer/utils.py
src/tensorcontainer.egg-info/PKG-INFO
src/tensorcontainer.egg-info/SOURCES.txt
src/tensorcontainer.egg-info/dependency_links.txt
src/tensorcontainer.egg-info/requires.txt
src/tensorcontainer.egg-info/top_level.txt
src/tensorcontainer/distributions/__init__.py
src/tensorcontainer/distributions/sampling.py
src/tensorcontainer/distributions/soft_bernoulli.py
src/tensorcontainer/distributions/symlog.py
src/tensorcontainer/distributions/truncated_normal.py
src/tensorcontainer/tensor_distribution/__init__.py
src/tensorcontainer/tensor_distribution/base.py
src/tensorcontainer/tensor_distribution/bernoulli.py
src/tensorcontainer/tensor_distribution/beta.py
src/tensorcontainer/tensor_distribution/binomial.py
src/tensorcontainer/tensor_distribution/categorical.py
src/tensorcontainer/tensor_distribution/cauchy.py
src/tensorcontainer/tensor_distribution/chi2.py
src/tensorcontainer/tensor_distribution/continuous_bernoulli.py
src/tensorcontainer/tensor_distribution/dirichlet.py
src/tensorcontainer/tensor_distribution/exponential.py
src/tensorcontainer/tensor_distribution/fisher_snedecor.py
src/tensorcontainer/tensor_distribution/gamma.py
src/tensorcontainer/tensor_distribution/geometric.py
src/tensorcontainer/tensor_distribution/gumbel.py
src/tensorcontainer/tensor_distribution/half_cauchy.py
src/tensorcontainer/tensor_distribution/half_normal.py
src/tensorcontainer/tensor_distribution/independent.py
src/tensorcontainer/tensor_distribution/inverse_gamma.py
src/tensorcontainer/tensor_distribution/kumaraswamy.py
src/tensorcontainer/tensor_distribution/laplace.py
src/tensorcontainer/tensor_distribution/lkj_cholesky.py
src/tensorcontainer/tensor_distribution/log_normal.py
src/tensorcontainer/tensor_distribution/logistic_normal.py
src/tensorcontainer/tensor_distribution/low_rank_multivariate_normal.py
src/tensorcontainer/tensor_distribution/mixture_same_family.py
src/tensorcontainer/tensor_distribution/multinomial.py
src/tensorcontainer/tensor_distribution/multivariate_normal.py
src/tensorcontainer/tensor_distribution/negative_binomial.py
src/tensorcontainer/tensor_distribution/normal.py
src/tensorcontainer/tensor_distribution/one_hot_categorical.py
src/tensorcontainer/tensor_distribution/one_hot_categorical_straight_through.py
src/tensorcontainer/tensor_distribution/pareto.py
src/tensorcontainer/tensor_distribution/poisson.py
src/tensorcontainer/tensor_distribution/relaxed_bernoulli.py
src/tensorcontainer/tensor_distribution/relaxed_one_hot_categorical.py
src/tensorcontainer/tensor_distribution/soft_bernoulli.py
src/tensorcontainer/tensor_distribution/student_t.py
src/tensorcontainer/tensor_distribution/symlog.py
src/tensorcontainer/tensor_distribution/tanh_normal.py
src/tensorcontainer/tensor_distribution/transformed_distribution.py
src/tensorcontainer/tensor_distribution/truncated_normal.py
src/tensorcontainer/tensor_distribution/uniform.py
src/tensorcontainer/tensor_distribution/utils.py
src/tensorcontainer/tensor_distribution/von_mises.py
src/tensorcontainer/tensor_distribution/weibull.py
src/tensorcontainer/tensor_distribution/wishart.py
tests/test_compile_utils.py
tests/test_types.py
tests/test_unsafe_construction.py