Kernels
kernels-bot commited on
Commit
38223fa
·
verified ·
1 Parent(s): 3f73c7a

Uploaded using `kernel-builder`.

Browse files
Files changed (1) hide show
  1. README.md +27 -0
README.md ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: kernels
3
+ license: mit
4
+ ---
5
+
6
+ This is the repository card of mlx-rmsnorm/mlx-rmsnorm 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("mlx-rmsnorm/mlx-rmsnorm")
15
+ layers = kernel_module.layers
16
+
17
+ layers(...)
18
+ ```
19
+
20
+ ## Available functions
21
+ - `layers`
22
+ - `rmsnorm_forward`
23
+ - `rmsnorm_backward`
24
+
25
+ ## Benchmarks
26
+
27
+ No benchmark available yet.