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
CustomOp | A declared custom op; call it eagerly or in a graph. |
|---|
Symbols
DTypeVar | A dtype variable: inputs sharing one must agree at call time. |
|---|---|
Symbol | A dim belonging to one CustomOp signature. |
TemplateType | One entry in a CustomOp signature, input or output. |
Symbols | Returns one Symbol per name, for tuple unpacking. |
|---|---|
declare | Declares a custom op from its signature. |