Python package
max
The MAX Python API reference.
The MAX API provides a high-performance graph compiler and runtime library that executes AI models with incredible speed on a wide range of hardware.
MAX offers a layered architecture that lets you work at the level of abstraction that best fits your needs. From deploying production-ready models with a few lines of code to building custom neural networks from scratch, each layer builds upon the others so you can move between levels seamlessly as requirements evolve.
For an introduction, see the Model developer guide.
Modules
- max.diagnostics.gpu
- max.driver
- max.dtype
- max.engine
- max.entrypoints
- max.graph
- max.graph.ops
- max.graph.quantization
- max.graph.weights
- max.interfaces
- max.kv_cache
- max.nn
- max.nn.attention
- max.nn.kernels
- max.nn.kv_cache
- max.pipelines
- max.pipelines.core
- max.pipelines.lib
- max.pipelines.lib.interfaces
- max.pipelines.lib.log_probabilities
- max.profiler
- max.experimental
- max.experimental.nn
- max.experimental.nn.norm
- max.experimental.nn.rope
- max.experimental.tensor
- max.experimental.random
- max.experimental.functional
- max.experimental.torch
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!
