Chapter 1 / 5
โฆ 0 XP
Chapter 1 / 5
Bolt on the toolkit
New here? Read the 60-sec briefing โThe fleet's standard AI toolkit is PyTorch. Bolt your Mojo conv op onto it so any crew member can call it from a torch model like a built-in, no Mojo knowledge required.
A PyTorch custom op registers your Mojo/MAX kernel under a torch namespace. Once bound, it flows through torch tensors and composes with the rest of a model, how production teams ship custom CUDA-class kernels.
YOUR TASK
- 1Register the op in the "fleet" namespace.
- 2Bind it as "fleet::conv1d".
๐ก @torch.library.custom_op("fleet::conv1d", mutates_args=())
ops/torch_conv1d.py
PUZZLE 20 ยท TORCH OPClear all 5 chapters to forge๐ฅ Fleet Standard Core Shard