Over the last several months, we’ve been progressively open-sourcing the Python API for MAX, from our AI models, to the MAX Graph API, to serving. Yesterday, we got everything in place to make the rest of this publicly available, so the entire MAX Python API is now open-source!
You can browse all of our Python code in the max directory in the modular repository. The newly-available modules include:
_core_core_mojo_core_types_mlirdriverdtypeengineexperimentalinterfacesmlirprofilersupporttorch
We’re in the process of wiring up the ability to test modifications to these in CI, which will make it practical for PRs to be made against them.