Funsor
0.4.2
Funsor Core:
Operations
Domains
Interpretations
Funsors
Optimizer
Adjoint Algorithms
Sum-Product Algorithms
Affine Pattern Matching
Funsor Factory
Testing Utiltites
Typing Utiltites
Interfaces:
Recipes using Funsor
Pyro-Compatible Distributions
Distribution Funsors
Mini-Pyro Interface
Einsum Interface
Compiler & Tracer
Tutorials and Examples
Named tensor notation with funsors (Part 1)
Example: Adam optimizer
Example: Discrete HMM
Example: Switching Linear Dynamical System EEG
Example: Forward-Backward algorithm
Example: Kalman Filter
Example: Mini Pyro
Example: PCFG
Example: Biased Kalman Filter
Example: Switching Linear Dynamical System
Example: Talbot’s method for numerical inversion of the Laplace transform
Example: VAE MNIST
Funsor
Docs
»
Index
Edit on GitHub
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
J
|
L
|
M
|
N
|
O
|
P
|
Q
|
R
|
S
|
T
|
U
|
V
|
W
|
X
|
Z
A
abs (in module funsor.ops.builtin)
abs() (Funsor method)
ActualExpected (class in funsor.testing)
Adam (class in funsor.minipyro)
add (in module funsor.ops.builtin)
adjoint() (AdjointTape method)
(in module funsor.adjoint)
adjoint_binary() (in module funsor.adjoint)
adjoint_cat() (in module funsor.adjoint)
adjoint_contract() (in module funsor.adjoint)
adjoint_contract_generic() (in module funsor.adjoint)
adjoint_contract_unary() (in module funsor.adjoint)
adjoint_reduce() (in module funsor.adjoint)
adjoint_scatter() (in module funsor.adjoint)
adjoint_subs() (in module funsor.adjoint)
AdjointTape (class in funsor.adjoint)
affine_inputs() (in module funsor.affine)
align() (Constant method)
(Contraction method)
(Delta method)
(Funsor method)
(Gaussian method)
(Tensor method)
align_gaussian() (in module funsor.gaussian)
align_tensor() (in module funsor.tensor)
align_tensors() (in module funsor.tensor)
all (in module funsor.ops.array)
all() (Funsor method)
amax (in module funsor.ops.array)
amin (in module funsor.ops.array)
and_ (in module funsor.ops.builtin)
any (in module funsor.ops.array)
any() (Funsor method)
apply_optimizer() (in module funsor.optimizer)
apply_stack() (in module funsor.minipyro)
Approximate (class in funsor.terms)
approximate() (Funsor method)
arg_constraints (FunsorDistribution attribute)
(GaussianHMM attribute)
(SwitchingLinearHMM attribute)
argmax (in module funsor.ops.array)
argmax() (Funsor method)
argmax_approximate (in module funsor.approximations)
argmin (in module funsor.ops.array)
argmin() (Funsor method)
arity (BinaryOp attribute)
(FinitaryOp attribute)
(NullaryOp attribute)
(Op attribute)
(TernaryOp attribute)
(UnaryOp attribute)
as_code() (OpProgram method)
as_tensor() (BlockMatrix method)
(BlockVector method)
assert_close() (in module funsor.testing)
assert_equiv() (in module funsor.testing)
astype (in module funsor.ops.array)
atanh (in module funsor.ops.builtin)
atanh() (Funsor method)
B
BernoulliLogits (class in funsor.torch.distributions)
BernoulliProbs (class in funsor.torch.distributions)
Beta (class in funsor.torch.distributions)
Binary (class in funsor.terms)
binary_divide() (in module funsor.cnf)
binary_subtract() (in module funsor.cnf)
binary_to_contract() (in module funsor.cnf)
BinaryOp (class in funsor.ops.op)
Binomial (class in funsor.torch.distributions)
Bint (class in funsor.domains)
bint() (in module funsor.domains)
BintType (class in funsor.domains)
block (class in funsor.minipyro)
BlockMatrix (class in funsor.gaussian)
BlockVector (class in funsor.gaussian)
Bound (class in funsor.factory)
C
CallableInterpretation (class in funsor.interpretations)
Cat (class in funsor.terms)
cat (in module funsor.ops.array)
Categorical (class in funsor.torch.distributions)
CategoricalLogits (class in funsor.torch.distributions)
Cauchy (class in funsor.torch.distributions)
check_funsor() (in module funsor.testing)
Chi2 (class in funsor.torch.distributions)
children_contraction() (in module funsor.cnf)
cholesky (in module funsor.ops.array)
cholesky_inverse (in module funsor.ops.array)
cholesky_solve (in module funsor.ops.array)
clamp (in module funsor.ops.array)
clamp_finite() (Tensor method)
ClippedAdam (class in funsor.minipyro)
combine_subs() (Precondition method)
compression_threshold (Gaussian attribute)
compute_argmax() (in module funsor.approximations)
CondIndepStackFrame (class in funsor.minipyro)
Constant (class in funsor.constant)
ConstantMeta (class in funsor.constant)
Contraction (class in funsor.cnf)
D
declare_op_types() (in module funsor.ops.op)
deep_isinstance() (in module funsor.typing)
deep_issubclass (in module funsor.typing)
deep_type() (in module funsor.typing)
default() (LogAbsDetJacobianOp static method)
(WrappedTransformOp static method)
Delta (class in funsor.delta)
(class in funsor.torch.distributions)
Dependent (class in funsor.domains)
DesugarGetitem (class in funsor.testing)
detach (in module funsor.ops.array)
diagonal (in module funsor.ops.array)
dim (CondIndepStackFrame attribute)
Dirichlet (class in funsor.torch.distributions)
DirichletMultinomial (class in funsor.torch.distributions)
DiscreteHMM (class in funsor.pyro.hmm)
DispatchedInterpretation (class in funsor.interpretations)
dispatcher (LogAbsDetJacobianOp attribute)
(WrappedTransformOp attribute)
dist_class (BernoulliLogits attribute)
(BernoulliProbs attribute)
(Beta attribute)
(Binomial attribute)
(Categorical attribute)
(CategoricalLogits attribute)
(Cauchy attribute)
(Chi2 attribute)
(Delta attribute)
(Dirichlet attribute)
(DirichletMultinomial attribute)
(Distribution attribute)
(Exponential attribute)
(Gamma attribute)
(GammaPoisson attribute)
(Geometric attribute)
(Gumbel attribute)
(HalfCauchy attribute)
(HalfNormal attribute)
(Laplace attribute)
(LowRankMultivariateNormal attribute)
(Multinomial attribute)
(MultivariateNormal attribute)
(NonreparameterizedBeta attribute)
(NonreparameterizedDirichlet attribute)
(NonreparameterizedGamma attribute)
(NonreparameterizedNormal attribute)
(Normal attribute)
(Pareto attribute)
(Poisson attribute)
(StudentT attribute)
(Uniform attribute)
(VonMises attribute)
dist_to_funsor() (in module funsor.pyro.convert)
distribute_subs_contraction() (in module funsor.cnf)
Distribution (class in funsor.distribution)
(class in funsor.minipyro)
do_fresh_subs() (in module funsor.cnf)
Domain (in module funsor.domains)
dtype (Bint attribute)
(Funsor attribute)
(RealsType attribute)
dynamic_partial_sum_product() (in module funsor.sum_product)
E
eager (in module funsor.interpretations)
eager_binary_constant_constant() (in module funsor.constant)
eager_binary_constant_tensor() (in module funsor.constant)
eager_binary_tensor_constant() (in module funsor.constant)
eager_contract_base() (in module funsor.optimizer)
eager_contraction_gaussian() (in module funsor.cnf)
eager_contraction_generic_recursive() (in module funsor.cnf)
eager_contraction_generic_to_tuple() (in module funsor.cnf)
eager_contraction_tensor() (in module funsor.cnf)
eager_contraction_to_binary() (in module funsor.cnf)
eager_contraction_to_reduce() (in module funsor.cnf)
eager_independent_joint() (in module funsor.joint)
eager_log_prob() (funsor.distribution.Distribution class method)
eager_markov_product() (in module funsor.sum_product)
eager_reduce() (Constant method)
(Delta method)
(Distribution method)
(Funsor method)
(Gaussian method)
(Stack method)
(Tensor method)
eager_reduce_add() (in module funsor.constant)
eager_reduce_exp() (in module funsor.joint)
eager_subs() (Cat method)
(Constant method)
(Delta method)
(Funsor method)
(Gaussian method)
(Independent method)
(MarkovProduct method)
(Scatter method)
(Slice method)
(Stack method)
(Tensor method)
(Variable method)
eager_unary() (Funsor method)
(Number method)
(Tensor method)
(in module funsor.constant)
einsum (in module funsor.ops.array)
einsum() (in module funsor.einsum)
Einsum() (in module funsor.tensor)
Elbo (class in funsor.elbo)
ELBO (class in funsor.minipyro)
elbo() (in module funsor.minipyro)
empty() (in module funsor.testing)
entropy() (Distribution method)
(Independent method)
enumerate_support() (Distribution method)
eq (in module funsor.ops.builtin)
excludes_backend() (in module funsor.testing)
exp (in module funsor.ops.builtin)
exp() (Funsor method)
expand (in module funsor.ops.array)
expand() (DiscreteHMM method)
(FunsorDistribution method)
(SwitchingLinearHMM method)
expand_inputs() (Distribution method)
Expectation() (in module funsor.minipyro)
Exponential (class in funsor.torch.distributions)
extract_affine() (in module funsor.affine)
F
filter() (SwitchingLinearHMM method)
find_domain() (in module funsor.domains)
finfo (in module funsor.ops.array)
FinitaryOp (class in funsor.ops.op)
flip (in module funsor.ops.array)
floordiv (in module funsor.ops.builtin)
forward_backward() (in module funsor.adjoint)
forward_filter_backward_precondition() (in module funsor.recipes)
forward_filter_backward_rsample() (in module funsor.recipes)
Fresh (class in funsor.factory)
full_like (in module funsor.ops.array)
Function (class in funsor.tensor)
function() (in module funsor.tensor)
Funsor (class in funsor.terms)
funsor.adjoint (module)
funsor.affine (module)
funsor.approximations (module)
funsor.cnf (module)
funsor.compiler (module)
funsor.constant (module)
funsor.delta (module)
funsor.domains (module)
funsor.einsum (module)
funsor.elbo (module)
funsor.factory (module)
funsor.gaussian (module)
funsor.integrate (module)
funsor.interpretations (module)
funsor.interpreter (module)
funsor.joint (module)
funsor.minipyro (module)
funsor.montecarlo (module)
funsor.ops.array (module)
funsor.ops.builtin (module)
funsor.ops.op (module)
funsor.ops.program (module)
funsor.ops.tracer (module)
funsor.optimizer (module)
funsor.precondition (module)
funsor.pyro.convert (module)
funsor.pyro.distribution (module)
funsor.pyro.hmm (module)
funsor.recipes (module)
funsor.sum_product (module)
funsor.tensor (module)
funsor.terms (module)
funsor.testing (module)
funsor.torch.distributions (module)
funsor.typing (module)
funsor_to_cat_and_mvn() (in module funsor.pyro.convert)
funsor_to_mvn() (in module funsor.pyro.convert)
funsor_to_tensor() (in module funsor.pyro.convert)
FunsorDistribution (class in funsor.pyro.distribution)
funsordistribution_to_funsor() (in module funsor.pyro.distribution)
G
Gamma (class in funsor.torch.distributions)
GammaPoisson (class in funsor.torch.distributions)
Gaussian (class in funsor.gaussian)
GaussianHMM (class in funsor.pyro.hmm)
GaussianMixture (in module funsor.cnf)
GaussianMRF (class in funsor.pyro.hmm)
ge (in module funsor.ops.builtin)
GenericTypeMeta (class in funsor.typing)
Geometric (class in funsor.torch.distributions)
get_args() (in module funsor.typing)
get_origin() (in module funsor.typing)
get_param_store() (in module funsor.minipyro)
get_trace() (trace method)
get_type_hints() (in module funsor.typing)
getitem (in module funsor.ops.builtin)
getslice (in module funsor.ops.builtin)
gt (in module funsor.ops.builtin)
Gumbel (class in funsor.torch.distributions)
H
HalfCauchy (class in funsor.torch.distributions)
HalfNormal (class in funsor.torch.distributions)
Has (class in funsor.factory)
has_enumerate_support (Distribution attribute)
has_rsample (DiscreteHMM attribute)
(GaussianHMM attribute)
(GaussianMRF attribute)
(SwitchingLinearHMM attribute)
I
id_from_inputs() (in module funsor.testing)
ignore_jit_warnings() (in module funsor.tensor)
Independent (class in funsor.terms)
input_vars (Funsor attribute)
Integrate (class in funsor.integrate)
interpret() (AdjointTape method)
Interpretation (class in funsor.interpretations)
interpretation() (in module funsor.interpreter)
inv (WrappedTransformOp attribute)
inv() (TransformOp static method)
invert (in module funsor.ops.builtin)
is_affine() (in module funsor.affine)
is_array() (in module funsor.testing)
is_full_rank (Gaussian attribute)
isnan (in module funsor.ops.array)
item() (Funsor method)
(Number method)
(Tensor method)
iter_subsets() (in module funsor.testing)
J
Jit (class in funsor.minipyro)
Jit_ELBO (class in funsor.minipyro)
JitTrace_ELBO() (in module funsor.minipyro)
JitTraceEnum_ELBO() (in module funsor.minipyro)
JitTraceMeanField_ELBO() (in module funsor.minipyro)
L
Lambda (class in funsor.terms)
Laplace (class in funsor.torch.distributions)
laplace_approximate (in module funsor.approximations)
lazy (in module funsor.interpretations)
le (in module funsor.ops.builtin)
lgamma (in module funsor.ops.builtin)
log (in module funsor.ops.builtin)
log() (Funsor method)
log1p (in module funsor.ops.builtin)
log1p() (Funsor method)
log_abs_det_jacobian (WrappedTransformOp attribute)
log_abs_det_jacobian() (TransformOp static method)
log_joint (class in funsor.minipyro)
log_normalizer (Gaussian attribute)
log_prob() (DiscreteHMM method)
(Distribution method)
(FunsorDistribution method)
(SwitchingLinearHMM method)
LogAbsDetJacobianOp (class in funsor.ops.op)
logaddexp (in module funsor.ops.array)
logsumexp (in module funsor.ops.array)
logsumexp() (Funsor method)
lower() (in module funsor.compiler)
LowRankMultivariateNormal (class in funsor.torch.distributions)
lshift (in module funsor.ops.builtin)
lt (in module funsor.ops.builtin)
M
make() (funsor.ops.op.Op class method)
make_chain_einsum() (in module funsor.testing)
make_einsum_example() (in module funsor.testing)
make_funsor() (in module funsor.factory)
make_hmm_einsum() (in module funsor.testing)
make_plated_hmm_einsum() (in module funsor.testing)
MarkovProduct (class in funsor.sum_product)
MarkovProductMeta (class in funsor.sum_product)
materialize() (Constant method)
(Tensor method)
matmul (in module funsor.ops.builtin)
matrix_and_mvn_to_funsor() (in module funsor.pyro.convert)
max (in module funsor.ops.builtin)
max() (Funsor method)
mean (in module funsor.ops.array)
mean() (Distribution method)
(Funsor method)
(Independent method)
mean_approximate (in module funsor.approximations)
Memoize (class in funsor.interpretations)
memoize() (in module funsor.interpretations)
Messenger (class in funsor.minipyro)
min (in module funsor.ops.builtin)
min() (Funsor method)
mixed_sequential_sum_product() (in module funsor.sum_product)
mod (in module funsor.ops.builtin)
modified_partial_sum_product() (in module funsor.sum_product)
moment_matching (in module funsor.interpretations)
moment_matching_contract_default() (in module funsor.joint)
moment_matching_contract_joint() (in module funsor.joint)
moment_matching_reduce() (Funsor method)
MonteCarlo (class in funsor.montecarlo)
mul (in module funsor.ops.builtin)
Multinomial (class in funsor.torch.distributions)
MultivariateNormal (class in funsor.torch.distributions)
mvn_to_funsor() (in module funsor.pyro.convert)
N
naive_contract_einsum() (in module funsor.einsum)
naive_einsum() (in module funsor.einsum)
naive_plated_einsum() (in module funsor.einsum)
naive_sarkka_bilmes_product() (in module funsor.sum_product)
naive_sequential_sum_product() (in module funsor.sum_product)
name (CondIndepStackFrame attribute)
(LogAbsDetJacobianOp attribute)
(WrappedTransformOp attribute)
ne (in module funsor.ops.builtin)
neg (in module funsor.ops.builtin)
new_arange (in module funsor.ops.array)
new_arange() (Tensor method)
new_eye (in module funsor.ops.array)
new_full (in module funsor.ops.array)
new_zeros (in module funsor.ops.array)
NonreparameterizedBeta (class in funsor.torch.distributions)
NonreparameterizedDirichlet (class in funsor.torch.distributions)
NonreparameterizedGamma (class in funsor.torch.distributions)
NonreparameterizedNormal (class in funsor.torch.distributions)
Normal (class in funsor.torch.distributions)
normalize (in module funsor.interpretations)
normalize_contraction_commutative_canonical_order() (in module funsor.cnf)
normalize_contraction_commute_joint() (in module funsor.cnf)
normalize_contraction_generic_args() (in module funsor.cnf)
normalize_contraction_generic_tuple() (in module funsor.cnf)
normalize_fuse_subs() (in module funsor.cnf)
normalize_trivial() (in module funsor.cnf)
null (in module funsor.ops.builtin)
NullaryOp (class in funsor.ops.op)
Number (class in funsor.terms)
O
of_shape() (in module funsor.terms)
ones() (in module funsor.testing)
Op (class in funsor.ops.op)
OpProgram (class in funsor.ops.program)
optimize_contract_finitary_funsor() (in module funsor.optimizer)
optimize_contraction_variadic() (in module funsor.optimizer)
or_ (in module funsor.ops.builtin)
P
param() (in module funsor.minipyro)
Pareto (class in funsor.torch.distributions)
partial_sum_product() (in module funsor.sum_product)
partial_unroll() (in module funsor.sum_product)
PatternMissingError
permute (in module funsor.ops.array)
plate() (in module funsor.minipyro)
PlateMessenger (class in funsor.minipyro)
Poisson (class in funsor.torch.distributions)
pop_interpretation() (in module funsor.interpreter)
pos (in module funsor.ops.builtin)
postprocess_message() (log_joint method)
(Messenger method)
(trace method)
pow (in module funsor.ops.builtin)
Precondition (class in funsor.precondition)
pretty() (Funsor method)
process_message() (block method)
(Messenger method)
(PlateMessenger method)
(log_joint method)
(replay method)
prod (in module funsor.ops.array)
prod() (Funsor method)
push_interpretation() (in module funsor.interpreter)
PyroOptim (class in funsor.minipyro)
Q
qr (in module funsor.ops.array)
quote() (Funsor method)
R
rand() (in module funsor.testing)
randint() (in module funsor.testing)
randn (in module funsor.ops.array)
randn() (in module funsor.testing)
random_gaussian() (in module funsor.testing)
random_mvn() (in module funsor.testing)
random_scale_tril() (in module funsor.testing)
random_tensor() (in module funsor.testing)
rank (Gaussian attribute)
Real (class in funsor.domains)
Reals (class in funsor.domains)
reals() (in module funsor.domains)
RealsType (class in funsor.domains)
reciprocal (in module funsor.ops.builtin)
Reduce (class in funsor.terms)
reduce() (Funsor method)
reduce_funsor() (in module funsor.cnf)
register() (Op method)
register_subclasscheck() (in module funsor.typing)
reinterpret() (in module funsor.interpreter)
replay (class in funsor.minipyro)
requires_backend() (in module funsor.testing)
requires_grad (Funsor attribute)
(Tensor attribute)
reshape() (Funsor method)
rsample() (FunsorDistribution method)
rshift (in module funsor.ops.builtin)
S
safediv (in module funsor.ops.builtin)
safesub (in module funsor.ops.builtin)
sample (in module funsor.ops.array)
sample() (Funsor method)
(FunsorDistribution method)
(in module funsor.minipyro)
sarkka_bilmes_product() (in module funsor.sum_product)
Scatter (class in funsor.terms)
scatter (in module funsor.ops.array)
scatter_add (in module funsor.ops.array)
seed (class in funsor.minipyro)
sequential (in module funsor.interpretations)
sequential_reduce() (Funsor method)
sequential_sum_product() (in module funsor.sum_product)
set_callable() (CallableInterpretation method)
set_compression_threshold() (funsor.gaussian.Gaussian class method)
set_inv() (TransformOp method)
set_log_abs_det_jacobian() (TransformOp method)
shape (Bint attribute)
(Funsor attribute)
(Real attribute)
(Reals attribute)
sigmoid (in module funsor.ops.builtin)
sigmoid() (Funsor method)
signature (LogAbsDetJacobianOp attribute)
(WrappedTransformOp attribute)
size (BintType attribute)
(CondIndepStackFrame attribute)
Slice (class in funsor.terms)
solve() (in module funsor.delta)
sqrt (in module funsor.ops.builtin)
sqrt() (Funsor method)
Stack (class in funsor.terms)
stack (in module funsor.ops.array)
StatefulInterpretation (class in funsor.interpretations)
std (in module funsor.ops.array)
std() (Funsor method)
step() (SVI method)
StudentT (class in funsor.torch.distributions)
sub (in module funsor.ops.builtin)
subclass_register() (funsor.ops.op.Op class method)
Subs (class in funsor.terms)
sum (in module funsor.ops.array)
sum() (Funsor method)
sum_product() (in module funsor.sum_product)
support (FunsorDistribution attribute)
SVI (class in funsor.minipyro)
SwitchingLinearHMM (class in funsor.pyro.hmm)
T
tanh (in module funsor.ops.builtin)
tanh() (Funsor method)
Tensor (class in funsor.tensor)
tensor_to_funsor() (in module funsor.minipyro)
(in module funsor.pyro.convert)
tensordot() (in module funsor.tensor)
TernaryOp (class in funsor.ops.op)
to_data() (in module funsor.terms)
to_funsor() (in module funsor.terms)
TorchOptimizer (Adam attribute)
(ClippedAdam attribute)
trace (class in funsor.minipyro)
Trace_ELBO (class in funsor.minipyro)
trace_function() (in module funsor.ops.tracer)
TraceEnum_ELBO (class in funsor.minipyro)
TraceMeanField_ELBO (class in funsor.minipyro)
TransformOp (class in funsor.ops.op)
transpose (in module funsor.ops.array)
triangular_inv (in module funsor.ops.array)
triangular_solve (in module funsor.ops.array)
truediv (in module funsor.ops.builtin)
typing_wrap (class in funsor.typing)
U
Unary (class in funsor.terms)
unary_contract() (in module funsor.cnf)
unary_log_exp() (in module funsor.cnf)
unary_neg_variable() (in module funsor.cnf)
UnaryOp (class in funsor.ops.op)
unfold_contraction_generic_tuple() (in module funsor.optimizer)
unfold_contraction_variadic() (in module funsor.optimizer)
Uniform (class in funsor.torch.distributions)
unsqueeze (in module funsor.ops.array)
V
var (in module funsor.ops.array)
var() (Funsor method)
Variable (class in funsor.terms)
Variadic (class in funsor.typing)
variance() (Distribution method)
(Independent method)
VonMises (class in funsor.torch.distributions)
W
WrappedTransformOp (class in funsor.ops.op)
X
xfail_if_not_found() (in module funsor.testing)
xfail_if_not_implemented() (in module funsor.testing)
xfail_param() (in module funsor.testing)
xor (in module funsor.ops.builtin)
Z
zeros() (in module funsor.testing)
Read the Docs
v: 0.4.2
Versions
latest
stable
0.4.2
0.4.1
0.4.0
0.3.0
0.2.0
0.1.2
0.1.1
0.1.0
Downloads
On Read the Docs
Project Home
Builds
Free document hosting provided by
Read the Docs
.