moso 0.0.1

No project description provided

moso

moso provides a %%moso cell magic that lets you build up a Mojo program across multiple notebook cells. All %%moso cells are collected, compiled, and run together as a single Mojo script. The last collected cell is wrapped in def main() raises:.

moso primarily supports SolveIt (from Answer.ai) because it allows the tool to identify which cell is currently executing. In a standard Jupyter Notebook environment, the magic simply collects all cells containing %%moso.

Install

pip install -e .

Usage

SolveIt

from moso.core import *
setup_moso()

Jupyter Notebook

from moso.core import *
setup_moso(nb_path='path/to/your/notebook.ipynb')

Then write Mojo code in %%moso cells:

%%moso
def add(a: Int, b: Int) -> Int:
    return a + b

%%moso
print(add(40, 2))

Remote execution

Pass an SSH host and command template if Mojo runs on another machine:

setup_moso(host="user@host", cmd="cd ~/mojo-gpu-puzzles && /root/.pixi/bin/pixi run mojo run {path}")

How it works

filter_moso strips the %%moso lines, concatenates previous cells, and wraps the current cell in a main() (or the last cell in standard Jupyter Notebook). setup_moso registers the IPython cell magic and runs the generated source either locally or over SSH.