Kernels:

---

drbh

/

yamoe

like 0

This is the repository card of drbh/yamoe that has been pushed on the Hub. It was built to be used with the kernels library. This card was automatically generated.

How to use

# make sure `kernels` is installed: `pip install -U kernels`
from kernels import get_kernel

kernel_module = get_kernel("drbh/yamoe")
ops = kernel_module.ops

ops(...)

Available functions

• ops
• Yamoe
• shuffle
• gather
• scatter
• sort
• bincount_cumsum
• batch_mm
• experts
• experts_backward
• reference
• yamoe_ref
• gpt_oss_mlp

Benchmarks

No benchmark available yet.