Note
This is a preview of the capability to access Mojo from Python code, and we plan to rapidly iterate on the interfaces and ergonomics used here.
Mojo now has the ability to be called from Python, with Mojo functions representing themselves as if they were native Python functions. When a Python script is run that references a Mojo module, that module is transparently compiled to present an interface that can be used from Python. At present, the interface that is presented to Python must be manually defined in Mojo code.
A full description of this functionality can be found within the Mojo manual, including current known limitations.
These examples show how to perform basic to more advanced use of Mojo code from Python in order to progressively replace hotspots in Python code with fast Mojo. This includes using Mojo to drive calculations on MAX-compatible GPUs.
The two examples of Mojo functions being called from Python consist of:
- hello.py: A literal "hello world" example, where a string is passed in from Python to Mojo, concatenated with a string in Mojo code, and the result passed back to Python to be printed.
- mandelbrot.py: A parallel calculation of the number of iterations to escape in the Mandelbrot set, performed on a MAX-compatible GPU in Mojo. The Python code calls into the Mojo GPU calculation and gets the results back as ASCII art in a string to be printed.
You can run these examples via Pixi:
pixi run hello
pixi run mandelbrotor directly in a Python virtual environment where the max PyPI package has
been installed:
python hello.py
python mandelbrot.py