Index

A

analyze() (effectful.ops.syntax.Scoped method) Annotation (class in effectful.ops.types) apply() (effectful.handlers.numpyro.Naming method) (effectful.handlers.pyro.Naming method) (in module effectful.ops.semantics)

args (effectful.ops.types.Term property)

B

BernoulliLogits() (in module effectful.handlers.numpyro) BernoulliProbs() (in module effectful.handlers.numpyro) Beta() (in module effectful.handlers.numpyro)

bind_dims() (in module effectful.handlers.jax) (in module effectful.handlers.torch) BinomialLogits() (in module effectful.handlers.numpyro) BinomialProbs() (in module effectful.handlers.numpyro)

C

canonicalize() (in module effectful.internals.unification) CategoricalLogits() (in module effectful.handlers.numpyro) CategoricalProbs() (in module effectful.handlers.numpyro) Cauchy() (in module effectful.handlers.numpyro)

Chi2() (in module effectful.handlers.numpyro) cond() (in module effectful.handlers.indexed) cond_n() (in module effectful.handlers.indexed) coproduct() (in module effectful.ops.semantics)

D

defdata() (in module effectful.ops.syntax) deffn() (in module effectful.ops.syntax) defop() (in module effectful.ops.syntax) defstream() (in module effectful.ops.syntax)

defterm() (in module effectful.ops.syntax) Delta() (in module effectful.handlers.numpyro) Dirichlet() (in module effectful.handlers.numpyro) DirichletMultinomial() (in module effectful.handlers.numpyro) Distribution() (in module effectful.handlers.numpyro)

E

effectful.handlers module effectful.handlers.indexed module effectful.handlers.jax module effectful.handlers.jax.numpy module effectful.handlers.jax.scipy module effectful.handlers.numpyro module effectful.handlers.pyro module effectful.handlers.torch module

effectful.internals module effectful.internals.runtime module effectful.internals.unification module effectful.ops module effectful.ops.semantics module effectful.ops.syntax module effectful.ops.types module evaluate() (in module effectful.ops.semantics) expand_to_batch_shape() (in module effectful.handlers.numpyro) Exponential() (in module effectful.handlers.numpyro)

F

freetypevars() (in module effectful.internals.unification) from_shape() (effectful.handlers.numpyro.Naming static method) (effectful.handlers.pyro.Naming static method)

fvsof() (in module effectful.ops.semantics) fwd() (in module effectful.ops.semantics)

G

Gamma() (in module effectful.handlers.numpyro) gather() (in module effectful.handlers.indexed) GeometricLogits() (in module effectful.handlers.numpyro) GeometricProbs() (in module effectful.handlers.numpyro)

get() (effectful.ops.types.Interpretation method) get_interpretation() (in module effectful.internals.runtime) get_runtime() (in module effectful.internals.runtime) grad() (in module effectful.handlers.torch) Gumbel() (in module effectful.handlers.numpyro)

H

HalfCauchy() (in module effectful.handlers.numpyro) HalfNormal() (in module effectful.handlers.numpyro)

handler() (in module effectful.ops.semantics) hessian() (in module effectful.handlers.torch)

I

implementations (effectful.ops.syntax.ObjectInterpretation attribute) implements() (in module effectful.ops.syntax) Independent() (in module effectful.handlers.numpyro) IndexSet (class in effectful.handlers.indexed) indices_of() (in module effectful.handlers.indexed) infer_annotations() (effectful.ops.syntax.Scoped class method) (effectful.ops.types.Annotation class method)

Interpretation (class in effectful.ops.types) interpretation (effectful.internals.runtime.Runtime attribute) interpreter() (in module effectful.internals.runtime) items() (effectful.ops.types.Interpretation method) iter_() (in module effectful.ops.syntax)

J

jacfwd() (in module effectful.handlers.torch) jacrev() (in module effectful.handlers.torch)

jax_getitem() (in module effectful.handlers.jax) jit() (in module effectful.handlers.jax) jvp() (in module effectful.handlers.torch)

K

keys() (effectful.ops.types.Interpretation method)

Kumaraswamy() (in module effectful.handlers.numpyro) kwargs (effectful.ops.types.Term property)

L

Laplace() (in module effectful.handlers.numpyro) LKJCholesky() (in module effectful.handlers.numpyro)

Logistic() (in module effectful.handlers.numpyro) LogNormal() (in module effectful.handlers.numpyro) LowRankMultivariateNormal() (in module effectful.handlers.numpyro)

M

module effectful.handlers effectful.handlers.indexed effectful.handlers.jax effectful.handlers.jax.numpy effectful.handlers.jax.scipy effectful.handlers.numpyro effectful.handlers.pyro effectful.handlers.torch effectful.internals effectful.internals.runtime effectful.internals.unification effectful.ops effectful.ops.semantics effectful.ops.syntax effectful.ops.types

MultinomialLogits() (in module effectful.handlers.numpyro) MultinomialProbs() (in module effectful.handlers.numpyro) MultivariateNormal() (in module effectful.handlers.numpyro)

N

name_to_sym() (in module effectful.handlers.indexed) Naming (class in effectful.handlers.numpyro) (class in effectful.handlers.pyro) NegativeBinomialLogits() (in module effectful.handlers.numpyro)

NegativeBinomialProbs() (in module effectful.handlers.numpyro) nested_type() (in module effectful.internals.unification) next_() (in module effectful.ops.syntax) Normal() (in module effectful.handlers.numpyro) NotHandled

O

ObjectInterpretation (class in effectful.ops.syntax) op (effectful.ops.types.Term property)

Operation (class in effectful.ops.types) ordinal (effectful.ops.syntax.Scoped attribute)

P

Pareto() (in module effectful.handlers.numpyro) Poisson() (in module effectful.handlers.numpyro) product() (in module effectful.ops.semantics)

pyro_module_shim() (in module effectful.handlers.pyro) pyro_sample() (in module effectful.handlers.pyro) PyroShim (class in effectful.handlers.pyro)

R

RelaxedBernoulliLogits() (in module effectful.handlers.numpyro)

runner() (in module effectful.ops.semantics) Runtime (class in effectful.internals.runtime)

S

Scoped (class in effectful.ops.syntax) sizesof() (in module effectful.handlers.jax) (in module effectful.handlers.torch)

stack() (in module effectful.handlers.indexed) StudentT() (in module effectful.handlers.numpyro) substitute() (in module effectful.internals.unification)

T

Term (class in effectful.ops.types) torch_getitem() (in module effectful.handlers.torch)

trace() (in module effectful.ops.syntax) typeof() (in module effectful.ops.semantics)

U

unbind_dims() (in module effectful.handlers.jax) (in module effectful.handlers.torch)

Uniform() (in module effectful.handlers.numpyro) unify() (in module effectful.internals.unification) union() (in module effectful.handlers.indexed)

V

values() (effectful.ops.types.Interpretation method) vjp() (in module effectful.handlers.torch)

vmap() (in module effectful.handlers.torch) VonMises() (in module effectful.handlers.numpyro)

W

Weibull() (in module effectful.handlers.numpyro)

Wishart() (in module effectful.handlers.numpyro)