Mojo Academy · 课时

MAX 图中的 Mojo

使用 Mojo 编写自定义操作

MAX 图中的 Mojo 是 CoddyKit 上的免费 Mojo Academy 课时。这是第 2 节课，共 4 节。你可以在下方免费阅读本课时的完整内容 — 然后在浏览器中使用内置代码编辑器和全天候 AI 导师进行实践。这是 Mojo Academy 学习路径的一部分，你的进度在网页和 CoddyKit 应用中同步。 Mojo Academy 课程共包含 4 节课。

本课时的部分内容尚未翻译，以英文显示。

Models Are Graphs

Inside MAX, a model is represented as a graph: a network of operations where data flows from one node to the next. 🔗

What an Op Is

Each node in the graph is an op, a single operation like a matrix multiply or an activation function applied to its inputs.

Built-in Ops

MAX ships with many built-in ops covering common layers, so most of a typical model graph works without any extra code.

The Need for Custom Ops

Sometimes your model needs an operation MAX does not include. That is when you write a custom op to fill the gap.

Mojo Powers Custom Ops

You write custom ops in Mojo, giving you systems-level speed for the exact computation your model requires.

An Op Is a Function

A custom op is essentially a Mojo function that takes input tensors and produces an output tensor for the graph.

fn my_op(x: Tensor) -> Tensor:

Speed Where It Counts

Because the op is Mojo, you can use SIMD and other tricks to make that node run as fast as the hardware allows.

Slotting Into the Graph

Once defined, your custom op is registered so MAX can place it in the graph just like any built-in operation.

Optimized Together

MAX still optimizes the full graph, including your custom node, fusing and scheduling work for the best overall throughput.

Best of Both Worlds

This pattern lets you keep a model mostly standard while dropping in a hand-tuned Mojo op exactly where you need speed.

When to Reach for It

Reach for a custom op when a missing or slow node is your bottleneck, not for parts the built-ins already handle well. ✨

Quick Check

Recall how Mojo connects to the MAX computation graph.

Recap

You saw that a MAX model is a graph of ops, and Mojo lets you write fast custom ops that slot right into it. 🎯

常见问题解答

「MAX 图中的 Mojo」课时是免费的吗？

是的 — 「MAX 图中的 Mojo」的完整文本可在网页上免费阅读。要进行交互式练习（内置代码编辑器和全天候 AI 导师）并解锁 Mojo Academy 课程的其余内容，请升级到 CoddyKit PRO。 Mojo Academy 课程共包含 4 节课。

「MAX 图中的 Mojo」这节课中我会学到什么？

使用 Mojo 编写自定义操作你通过在浏览器中直接运行的动手代码来练习 Mojo Academy，全天候 AI 导师会在你学习这节课的过程中回答你的问题。

学习 Mojo Academy 需要有经验吗？

无需任何先前经验。CoddyKit 上的 Mojo Academy 课程适合初学者到高级学习者，你可以从这里开始或从头开始，按照自己的节奏学习。这是第 2 节课，共 4 节。

「MAX 图中的 Mojo」课时需要多长时间？

大多数 CoddyKit 课程大约需要 5–10 分钟。每节课都很精短且互动，所以你能稳步进步，并在网页和应用中从离开的地方继续。

我能在这节 Mojo Academy 课中编写并运行代码吗？

能。每节 Mojo Academy 课都包含内置代码编辑器，你可以在浏览器中直接编写并运行真实代码，并获得即时 AI 反馈 — 无需本地设置。

此课程中的所有课时

← 返回 Mojo Academy