text-generation-inference/server/text_generation_server/layers/moe
Daniël de Kok 29a93b78ba Move to moe-kernels package and switch to common MoE layer (#2511)
* 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.

* Make `cargo check` pass

* Update runner
2024-09-25 06:18:05 +00:00
..
__init__.py Move to moe-kernels package and switch to common MoE layer (#2511) 2024-09-25 06:18:05 +00:00
unquantized.py Move to moe-kernels package and switch to common MoE layer (#2511) 2024-09-25 06:18:05 +00:00