Kernels
kernels-bot commited on
Commit
29d7ea4
·
verified ·
1 Parent(s): 83d306d

Uploaded using `kernel-builder`.

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -12,13 +12,12 @@ This is the repository card of mlx-rmsnorm/mlx-rmsnorm that has been pushed on t
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
 
 
12
  from kernels import get_kernel
13
 
14
  kernel_module = get_kernel("mlx-rmsnorm/mlx-rmsnorm")
15
+ rmsnorm_forward = kernel_module.rmsnorm_forward
16
 
17
+ rmsnorm_forward(...)
18
  ```
19
 
20
  ## Available functions
 
21
  - `rmsnorm_forward`
22
  - `rmsnorm_backward`
23