

Bernoulli


Beta


Binomial


Categorical


Cauchy


Chi2


ContinuousBernoulli


Dirichlet


Exponential


FisherSnedecor


Gamma


Geometric


Gumbel


HalfCauchy


HalfNormal


Independent


Kumaraswamy


LKJCholesky


Laplace


LogNormal


LowRankMultivariateNormal


MixtureSameFamily


Multinomial


MultivariateNormal


NegativeBinomial


Normal


OneHotCategorical


Pareto


Poisson


RelaxedBernoulli


LogitRelaxedBernoulli


RelaxedOneHotCategorical


StudentT


TransformedDistribution


Uniform


VonMises


Weibull


Wishart


KL Divergence


Transforms


Constraints


Constraint Registry


torch._dynamo


torch.fft


Fast Fourier Transforms


Helper Functions


torch.func


What are composable function transforms?


Why composable function transforms?


Read More


torch.futures


torch.fx


Overview


Writing Transformations


Debugging

