Commit Graph

1 Commits

Author SHA1 Message Date
Daniël de Kok
5726a9ca81 Move to moe-kernels package and switch to common MoE layer
This change introduces the new `moe-kernels` package:

- Add `moe-kernels` as a dependency.
- Introduce a `SparseMoELayer` module that can be used by MoE
  models.
- Port over Mixtral and Deepseek.
2024-09-16 10:57:44 +00:00