MaverickJune/pMoE

Sub token level load balancing of MoE models

0

stars

2

commits

Python

primary language

Feb 20, 2025

updated

README

pMoE

Predictable MoE that fully balanced the expert workloads

Table of Contents

Installation

Provide instructions on how to install the project. For example:

# Clone the repository
git https://github.com/MaverickJune/pMoE.git

# Navigate to the project directory
cd fastmoe

# Install dependencies
python setup.py install

Usage

Replace your ffn layers in the mode to pMoE layers

License

This project is licensed under the MIT License.

Contributors

MaverickJune

1 commits

sanghyun-snu

1 commits

MaverickJune/pMoE

Sub token level load balancing of MoE models

0

stars

2

commits

Python

primary language

Feb 20, 2025

updated

README

pMoE

Predictable MoE that fully balanced the expert workloads

Table of Contents

Installation

Provide instructions on how to install the project. For example:

# Clone the repository
git https://github.com/MaverickJune/pMoE.git

# Navigate to the project directory
cd fastmoe

# Install dependencies
python setup.py install

Usage

Replace your ffn layers in the mode to pMoE layers

License

This project is licensed under the MIT License.

Contributors

MaverickJune

1 commits

sanghyun-snu

1 commits

Languages

Python

81.3%

C++

10.0%

Cuda

3.9%

Shell

3.6%

C

1.1%