Skip to content

effectful.rst

Latest commit

 

History

History
104 lines (75 loc) · 2.26 KB

effectful.rst

File metadata and controls

104 lines (75 loc) · 2.26 KB
Raw

Effectful

Operations

.. automodule:: effectful.ops
   :members:
   :undoc-members:

Syntax

.. automodule:: effectful.ops.syntax
   :members:
   :undoc-members:

   .. autofunction:: effectful.ops.syntax.deffn(body: T, *args: Operation, **kwargs: Operation) -> Callable[..., T])
   .. autofunction:: effectful.ops.syntax.defterm(value: T) -> Expr[T]
   .. autofunction:: effectful.ops.syntax.defdata(value: Term[T]) -> Expr[T]
   .. autofunction:: effectful.ops.semantics.fwd

Semantics

.. automodule:: effectful.ops.semantics
   :members:
   :undoc-members:

   .. autofunction:: effectful.ops.semantics.apply

Types

.. automodule:: effectful.ops.types
   :members:
   :undoc-members:

Handlers

.. automodule:: effectful.handlers
   :members:
   :undoc-members:

Numbers

.. automodule:: effectful.handlers.numbers
   :members:
   :undoc-members:

Pyro

.. automodule:: effectful.handlers.pyro
   :members:
   :undoc-members:

   .. autofunction:: effectful.handlers.pyro.pyro_sample(name: str, fn: pyro.distributions.torch_distribution.TorchDistributionMixin, *args, obs: Optional[torch.Tensor] = None, obs_mask: Optional[torch.BoolTensor] = None, mask: Optional[torch.BoolTensor] = None, infer: Optional[pyro.poutine.runtime.InferDict] = None, **kwargs) -> torch.Tensor

Torch

.. automodule:: effectful.handlers.torch
   :members:
   :undoc-members:

   .. autofunction:: effectful.handlers.torch.grad
   .. autofunction:: effectful.handlers.torch.jacfwd
   .. autofunction:: effectful.handlers.torch.jacrev
   .. autofunction:: effectful.handlers.torch.hessian
   .. autofunction:: effectful.handlers.torch.jvp
   .. autofunction:: effectful.handlers.torch.vjp
   .. autofunction:: effectful.handlers.torch.vmap
   .. autofunction:: torch_getitem(x: torch.Tensor, key: Tuple[IndexElement, ...]) -> torch.Tensor

Indexed

.. automodule:: effectful.handlers.indexed
   :members:
   :undoc-members:

Internals

.. automodule:: effectful.internals
   :members:
   :undoc-members:


.. automodule:: effectful.internals.base_impl
   :members:
   :undoc-members:


.. automodule:: effectful.internals.runtime
   :members:
   :undoc-members:


.. automodule:: effectful.internals.unification
   :members:
   :undoc-members: