Python module
max.experimental.nn
Module implementation using eager tensors.
Base classes
Module | The core unit of composition for modeling in MAX. |
|---|---|
ModuleList | A Module subclass which is locally a list container. |
Sequential | A Module subclass which holds a sequence of unary modules. |
Linear layers
Conv2d | A 2D convolution layer. |
|---|---|
Embedding | A vector embedding. |
Linear | A unary linear transformation over an input tensor. |
Functions
module_dataclass | Converts a class into a MAX module with automatic parameter tracking. |
|---|
Submodules
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!