python_modules

Build custom neural network modules with MAX Python API

Examples demonstrating how to build custom neural network modules using the MAX Python API.

To run this example, execute the following command:

python main.py

The following is the expected output:

--- Simple MLP Block ---
MLPBlock(1 linear layers, 1 activations)
------------------------------
--- MLP Block (1 Hidden Layer) ---
MLPBlock(2 linear layers, 1 activations)
------------------------------
--- Deeper MLP Block (3 Hidden Layers, GELU) ---
MLPBlock(4 linear layers, 3 activations)
------------------------------