IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /get-started.md). For the complete documentation index, see llms.txt.
Skip to main content
For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Python module

max.experimental.custom

User-facing definitions of custom ops for eager and graph use.

CustomOp stages a mo.custom op from a declared inputs/outputs signature; works both eagerly (Tensor) and inside a Graph build (TensorValue). Interpreter binding logic lives in custom_gc.

Custom operations​

CustomOpA declared custom op; call it eagerly or in a graph.

Symbols​

DTypeVarA dtype variable: inputs sharing one must agree at call time.
SymbolA dim belonging to one CustomOp signature.
TemplateTypeOne entry in a CustomOp signature, input or output.
SymbolsReturns one Symbol per name, for tuple unpacking.
declareDeclares a custom op from its signature.