Kernels:

---

kernels-staging

/

sonic-moe

like 0

Follow

Kernels Staging 6

Trusted publisher

---
library_name: kernels
license: apache-2.0
---

This is the repository card of kernels-community/sonic-moe that has been pushed on the Hub. It was built to be used with the [`kernels` library](https://github.com/huggingface/kernels). This card was automatically generated.

## How to use

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

kernel_module = get_kernel("kernels-community/sonic-moe", version=2)
KernelBackendMoE = kernel_module.KernelBackendMoE

KernelBackendMoE(...)
```

## Available functions
- `KernelBackendMoE`
- `MoE`
- `moe_general_routing_inputs`
- `moe_TC_softmax_topk_layer`

## Benchmarks

No benchmark available yet.