Kernels
kernels-bot commited on
Commit
b40727b
·
verified ·
1 Parent(s): 8c5b0c3

Uploaded using `kernel-builder`.

Browse files
Files changed (1) hide show
  1. README.md +28 -3
README.md CHANGED
@@ -1,5 +1,30 @@
1
- # ScatterMoE
 
 
 
2
 
3
- This is the [ScatterMoE](https://arxiv.org/abs/2403.08245) written in Triton
4
 
5
- The main repository for ScatterMoE is [here](https://github.com/shawntan/scattermoe)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: kernels
3
+ license: apache-2.0
4
+ ---
5
 
6
+ This is the repository card of kernels-community/scattermoe 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.
7
 
8
+ ## How to use
9
+
10
+ ```python
11
+ # make sure `kernels` is installed: `pip install -U kernels`
12
+ from kernels import get_kernel
13
+
14
+ kernel_module = get_kernel("kernels-community/scattermoe")
15
+ flatten_sort_count = kernel_module.flatten_sort_count
16
+
17
+ flatten_sort_count(...)
18
+ ```
19
+
20
+ ## Available functions
21
+ - `flatten_sort_count`
22
+ - `parallel_linear`
23
+ - `ParallelExperts`
24
+ - `parallel_experts`
25
+ - `kernels`
26
+ - `layers`
27
+
28
+ ## Benchmarks
29
+
30
+ No benchmark available yet.