MXNet 2.0 API reference¶
This is the Python API reference for the smolix/mxnet fork — a port of Apache MXNet 2.0 to CUDA 13 / Blackwell GPUs and native Apple Silicon CPU.
It is generated directly from the docstrings in the installed mxnet package
(version 2.0.0+cpu.linux.20260617.1). There is no hand-written narrative or tutorial content; for
install, build, fork changes, and known issues see the repository’s
README.md, FIXED.md, and OPEN_ISSUES.md.
A warning to anyone wanting to use it: do so at your own risk. This is a port to ensure that some legacy packages, such as D2L can still execute mxnet code on hardware that isn’t ancient. You should not build new software using MXNet but it might greatly simplify your life if you have an old library that you need to maintain.
Note
mxnet.np / mxnet.npx are runtime aliases for the modules documented
below as mxnet.numpy and mxnet.numpy_extension.
Packages and modules¶
MXNet NumPy module. |
|
Module for ops not belonging to the official numpy package for imperative programming. |
|
NDArray API of MXNet. |
|
Symbol API of MXNet. |
|
Neural network module. |
|
Optimizer API of MXNet. |
|
Scheduling learning rate. |
|
Data iterators for common data formats and utility functions. |
|
Image Iterators and image augmentation functions |
|
Autograd for NDArray. |
|
Key-value store for distributed communication |
|
Device management API of mxnet. |
|
Profiler setting methods. |
|
Runtime querying of compile time features in the native library. |
|
Experimental contributions |