Module dopt.core

This package contains the framework for constructing and executing operation graphs.

  • dopt.core.ops provides functions for constructing nodes in the operation graph.
  • dopt.core.grads provides functions for computing the derivatives of operations.
  • dopt.core.cpu contains a backend that executes operation graphs using the CPU.
  • dopt.core.cuda contains a backend that executes operation graphs using a CUDA enabled GPU.

Functions

NameDescription
compile(outputs)Compile an Operation graph into a reusable execution plan.
evaluate(op, args)Evaluates an operation graph with a single root node.
evaluate(ops, args)Evaluates a several nodes from the operation graph.