Nx.Defn.Evaluator (Nx v0.13.0)

Copy Markdown View Source

The default implementation of a Nx.Defn.Compiler that evaluates the expression tree against the tensor backend.

Options

The following options are specific to this compiler:

  • :hooks - a map of callbacks to override named hook side effects. This allows overriding named hooks at JIT time without recompiling the graph.

  • :garbage_collect - when true, garbage collects after evaluating each node

  • :max_concurrency - the number of partitions to start when running a Nx.Serving with this compiler