Tutel

KO | EN

gitlite — search

#javascript #python #ai #typescript #llm #golang #rust #docker #java #windows #cli #kubernetes #openai #mcp

(★ 1,010)

Tutel MoE: Optimized Mixture-of-Experts Library, Support GptOss/DeepSeek/Kimi-K2/Qwen3 using FP8/NVFP4/MXFP4

File Explorer

Download Latest Version (.zip)

- Asymmetric-RMS-Norms.svg
- CHECKPOINT.md
- DeepSeek-V3.2.png
- DeepSeekR1-tutel-accel.png
- test_baseline.json
- test_tutel.py
- - __init__.py
  - gather.py
  - scatter.py
- - __init__.py
  - antares_ops.h
  - backend.hpp
  - custom_kernel.cpp
- - __init__.py
  - bandwidth_test.py
  - figure.svg
  - helloworld.py
  - helloworld_amp.py
  - helloworld_custom_expert_sharded.py
  - helloworld_custom_gate_expert.py
  - helloworld_ddp.py
  - helloworld_ddp_tutel.py
  - helloworld_from_scratch.py
  - helloworld_switch.py
  - moe_cifar10.py
  - moe_mnist.py
  - nccl_all_gather_v.py
  - nccl_all_to_all_v.py
  - nccl_allreduce_perf.py
  - oai_request.py
  - oai_request_stream.py
  - README.md
- - __init__.py
  - ffn.py
  - llama_ffn.py
- - __init__.py
  - cosine_top.py
  - top.py
- - __init__.py
  - communicate.py
  - fast_dispatch.py
  - jit_compiler.py
  - losses.py
  - moe_layer.py
  - overlap.py
- - __init__.py
  - gating.py
  - sparse.py
- - __init__.py
  - execl.py
  - run.py
- - __init__.py
- - __init__.py
  - patterns.py
  - solver.py
  - spmdx.py
- __init__.py
- jit.py
- moe.py
- net.py
- system.py
.gitignore
CODE_OF_CONDUCT.md
figure.svg
LICENSE
MANIFEST.in
README.md
SECURITY.md
setup.py
SUPPORT.md

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Tutel

Tutel MoE: An Optimized Mixture-of-Experts Implementation, also the first parallel solution proposing ["No-penalty Parallism/Sparsity/Capacity/.. Switching"](https://mlsys.org/media/mlsys-2023/Slides/2477.pdf) for modern training and inference that have dynamic behaviors. Support direct NVFP4/MXFP4/BlockwiseFP8 Inference for MoE-based GLM-5.x / DeepSeek-3.x / Kimi-2.x / Kimi-3.x / Qwen3 / Gpt-OSS using A100/A800/H100/MI300/..

> [!TIP]
> #### Steps for Kimi-K3/GLM-5.x (Claude-Code Mode):
>
> ☑ A100x8/H100x8 (80G SXM) for GLM-5.x (0.8T): max-context-size = 1M
> 
> ☑ MI300x8 (192GB PCIe5) for GLM-5.x (0.8T): max-context-size = 1M
> 
> ☑ MI300x8 (192GB PCIe5) for Kimi K3 (2.8T): max-context-size = 1M
> 
> | Azure GPU Type (x8) | ***vLLM/SGL Kimi K3*** (no-MTP) | ***Tutel Kimi K3*** (no-MTP) |
> |  ----  | ----  | ----  |
> | AMD MI300X 2023 (750W x8) | 0 t/s (OoM) | 73.4 t/s |
> | AMD MI325X 2024 (1000W x8) | 3.1 t/s  | 82.0 t/s |
> | AMD MI355X 2025 (1400W x8) | 43.5 t/s | (TBD, no environment available) |
> | NVIDIA B200 2024 (1000W x8) | 0 t/s (OoM)  | (TBD, no environment available) |
> 
> |  | FP4 + No Tools | FP4 + Tool Use |
> |  ----  | ----  | ----  |
> | Kimi-K3 @ GSM8K | 96.9%  | 97.3% |
> | GLM-5.2 @ GSM8K | 97.1%  | 97.8% |
> | GLM-5.2 @ AIME25 | 96.1%  | 99.7% |
> | GLM-5.2 @ AIME26 | 93.5%  | 99.6% |
>
> ***K3 Known Issues***: *(1) Kimi K3 support for MI300x/MI325x instead of A100x8 (GLM-5 only); (2) Limited by device memory, Kimi K3 only enables bsz=1 for MI300x8 (to be fixed soon);*
> 
> ```sh
> 
> [Model Downloads]
>   pip3 install -U "huggingface_hub[cli]" --upgrade
>   hf download --local-dir moonshotai/Kimi-K3 moonshotai/Kimi-K3
>   hf download --local-dir nvidia/GLM-5.2-NVFP4 nvidia/GLM-5.2-NVFP4
>   hf download --local-dir nvidia/GLM-5.1-NVFP4 nvidia/GLM-5.1-NVFP4
>   hf download --local-dir nvidia/GLM-5-NVFP4 nvidia/GLM-5-NVFP4
>
> [ND_MI300_192G_v5: Serve Kimi-K3/GLM-5/5.1/5.2 (for Azure MI300x8 PCIe)]
>   docker run -e WORKER=1 -e LOCAL_SIZE=8 -p 8000:8000 -it --rm --ipc=host --shm-size=8g \
>       --ulimit memlock=-1 --ulimit stack=67108864 -v /:/host -w /host$(pwd) \
>       --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --device=/dev/kfd --device=/dev/dri --group-add=video \
>       tutelgroup/deepseek-671b:mi300x8-chat-20260808 --serve=core \
>         --try_path moonshotai/Kimi-K3 \
>         --try_path nvidia/GLM-5.2-NVFP4 \
>         --try_path nvidia/GLM-5.1-NVFP4 \
>         --try_path nvidia/GLM-5-NVFP4 \
>         --max_seq_len 120000
>
> [ND_A100_80G_v4: Serve GLM-5/5.1/5.2 (for Azure A100x8/H100x8/B200x8 SXM)]
>   docker run -e WORKER=1 -e LOCAL_SIZE=8 -p 8000:8000 -it --rm --ipc=host --shm-size=8g \
>       --ulimit memlock=-1 --ulimit stack=67108864 -v /:/host -w /host$(pwd) \
>       -v /usr/lib/x86_64-linux-gnu/libcuda.so.1:/usr/lib/x86_64-linux-gnu/libcuda.so.1 --privileged \
>       tutelgroup/deepseek-671b:a100x8-chat-20260707 --serve=core \
>         --try_path nvidia/GLM-5.2-NVFP4 \
>         --max_seq_len 1000000
> ```
> 
> #### Setup Claude Code for Linux / WSL (Ubuntu >= 24.04):
> ```sh
> sudo apt-get install -y npm
> sudo npm install -g @anthropic-ai/claude-code@2.1.197
> cat > run_claude.sh <<EOF && chmod a+x run_claude.sh
> mkdir -p config/
> export ANTHROPIC_BASE_URL="http://0.0.0.0:8000"
> export ANTHROPIC_API_KEY="sk-ant-api00-local-mock-key"
> export CLAUDE_CONFIG_DIR="config"
> export DISABLE_AUTOUPDATER=1
> echo '{"customApiKeyResponses": {"approved": ["api00-local-mock-key"]}}' > config/.claude.json
> claude
> EOF
>
> ./run_claude.sh
> ```
> 
> #### Setup Claude Code for Windows (>= 10.0):
> ```sh
> winget install OpenJS.NodeJS.LTS
> winget install --id Git.Git -e --source winget
> npm install -g @anthropic-ai/claude-code@2.1.197
>   (
>     echo(@echo off
>     echo(if not exist config mkdir config
>     echo(set ANTHROPIC_BASE_URL=http://0.0.0.0:8000
>     echo(set ANTHROPIC_API_KEY=sk-ant-api00-local-mock-key
>     echo(set CLAUDE_CONFIG_DIR=config
>     echo(set DISABLE_AUTOUPDATER=1
>     echo(echo({"customApiKeyResponses": {"approved": ["api00-local-mock-key"]}} ^> config\.claude.json
>     echo(claude
> ) > run_claude.bat
>
> .\run_claude.bat
> ```
------------------

> [!TIP]
> #### Steps for Kimi-K2.6/2.7/DeepSeek V3.2 (Long-Context Mode):
> 
> ```sh
> [Model Downloads]
>   pip3 install -U "huggingface_hub[cli]" --upgrade
>   hf download moonshotai/Kimi-K2.7-Code --local-dir moonshotai/Kimi-K2.7-Code
>   hf download moonshotai/Kimi-K2.6 --local-dir moonshotai/Kimi-K2.6
>   hf download nvidia/Kimi-K2.5-NVFP4 --local-dir nvidia/Kimi-K2.5-NVFP4
>   hf download nvidia/Kimi-K2-Thinking-NVFP4 --local-dir nvidia/Kimi-K2-Thinking-NVFP4
>   hf download nvidia/DeepSeek-V3.2-NVFP4 --local-dir nvidia/DeepSeek-V3.2-NVFP4
> 
> [DeepSeek V3.2 Long-Context (for Azure A100x8/H100x8/B200x8 SXM)]
>   docker run -e LOCAL_SIZE=8 -e WORKER=1 -it --rm --ipc=host --net=host --shm-size=8g \
>       --ulimit memlock=-1 --ulimit stack=67108864 -v /:/host -w /host$(pwd) -v /tmp:/tmp \
>       -v /usr/lib/x86_64-linux-gnu/libcuda.so.1:/usr/lib/x86_64-linux-gnu/libcuda.so.1 --privileged \
>       tutelgroup/deepseek-671b:a100x8-chat-20260707 --serve=webui --listen_port 8000 \
>         --try_path moonshotai/Kimi-K2.7-Code \
>         --try_path moonshotai/Kimi-K2.6 \
>         --try_path nvidia/Kimi-K2.5-NVFP4 \
>         --try_path nvidia/Kimi-K2-Thinking-NVFP4 \
>         --try_path nvidia/DeepSeek-V3.2-NVFP4 \
>         --try_path nvidia/DeepSeek-R1-NVFP4 \
>         --max_seq_len 16384
> 
> [DeepSeek V3.2 Long-Context (for Azure MI300x8 PCIe)]
>   docker run -e LOCAL_SIZE=8 -e WORKER=1 -it --rm --ipc=host --net=host --shm-size=8g \
>       --ulimit memlock=-1 --ulimit stack=67108864 --device=/dev/kfd --device=/dev/dri --group-add=video \
>       --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /:/host -w /host$(pwd) -v /tmp:/tmp \
>       tutelgroup/deepseek-671b:mi300x8-chat-20260707 --serve=webui --listen_port 8000 \
>         --try_path moonshotai/Kimi-K2.7-Code \
>         --try_path moonshotai/Kimi-K2.6 \
>         --try_path nvidia/Kimi-K2.5-NVFP4 \
>         --try_path nvidia/Kimi-K2-Thinking-NVFP4 \
>         --try_path nvidia/DeepSeek-V3.2-NVFP4 \
>         --try_path nvidia/DeepSeek-R1-NVFP4 \
>         --max_seq_len 1000000
> 
> [OpenAI/Ollama/Direct Request]
>   curl -N -X POST http://0.0.0.0:8000/chat -d '{"text": "Write a Python code of the Quicksort algorithm."}'
>   python3 -m tutel.examples.oai_request_stream --url '0.0.0.0:8000' --prompt 'Write a Python code of the Quicksort algorithm.'
> 
> [Open-WebUI URL for Web browsers]
>   xdg-open http://0.0.0.0:8000
> ```

------------------
> [!TIP]
> #### Steps for Microsoft VibeVoice (Multimodality Mode):
> 
> ```sh
> [Model Downloads]
>   pip3 install -U "huggingface_hub[cli]" --upgrade
>   hf download microsoft/VibeVoice-1.5B --local-dir microsoft/VibeVoice-1.5B
>   hf download Qwen/Qwen2.5-1.5B --local-dir Qwen/Qwen2.5-1.5B
>
>   hf download aoi-ot/VibeVoice-Large --local-dir microsoft/VibeVoice-Large
>   hf download Qwen/Qwen2.5-7B --local-dir Qwen/Qwen2.5-7B
> 
> [Microsoft VibeVoice (ND_A100/H100/B200 only)]
> docker run -e LOCAL_SIZE=1 -it --rm -p 8001:8000 --shm-size=8g \
>       --ulimit memlock=-1 --ulimit stack=67108864 -v /:/host -w /host$(pwd) -v /tmp:/tmp \
>       -v /usr/lib/x86_64-linux-gnu/libcuda.so.1:/usr/lib/x86_64-linux-gnu/libcuda.so.1 --privileged \
>       -e VOICES="https://homepages.inf.ed.ac.uk/htang2/notes/speech-samples/103-1240-0000.wav" \
>       tutelgroup/deepseek-671b:a100x8-chat-20251222 --serve=core \
>         --try_path ./microsoft/VibeVoice-1.5B \
>         --try_path ./microsoft/VibeVoice-Large
> 
> [Microsoft VibeVoice (ND_MI300_192G_v5  only)]
> docker run -e LOCAL_SIZE=1 -it --rm -p 8001:8000 --shm-size=8g \
>       --ulimit memlock=-1 --ulimit stack=67108864 --device=/dev/kfd --device=/dev/dri --group-add=video \
>       --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /:/host -w /host$(pwd) -v /tmp:/tmp \
>       -e VOICES="https://homepages.inf.ed.ac.uk/htang2/notes/speech-samples/103-1240-0000.wav" \
>       tutelgroup/deepseek-671b:mi300x8-chat-20251222 --serve=core \
>         --try_path ./microsoft/VibeVoice-1.5B \
>         --try_path ./microsoft/VibeVoice-Large
> 
> [Audio Generation Request]
>   curl -X POST http://0.0.0.0:8001/chat -d '{"text": "VibeVoice is a novel framework designed for generating expressive, long-form, multi-speaker conversational audio, such as podcasts, from text."}' > sound_output.mp3
> ```

#### Inference TPOS for DeepSeek-MoE/Qwen3-MoE/KimiK2-MoE/GptOSS-MoE/..:
> |  ***Model \& Machine Type*** | ***Precision*** | ***SGL*** (no-MTP) | ***Tutel*** (no-MTP) |
> |  ----  | ----  | ----  | ----  |
> | $deepseek-ai/DeepSeek-V3.2\ (671B,\ A100 \times 8)$ | nvfp4 | - | 102 |
> | $deepseek-ai/DeepSeek-V3.2\ (671B,\ MI300 \times 8)$ | nvfp4 | - | 151 |
> | $moonshotai/Kimi-K2-Instruct\ (1T,\ A100 \times 8)$ | nvfp4 | - | 104 |
> | $moonshotai/Kimi-K2-Instruct\ (1T,\ MI300 \times 8)$ | fp8b128 | 49 | 153 |
> | $NVFP4/Qwen3-235B-A22B-Instruct-2507-FP4(A100\times8)$ | nvfp4 | - | 114 |
> | $NVFP4/Qwen3-235B-A22B-Instruct-2507-FP4(MI300\times8)$ | nvfp4 | - | 122 |
> | $openai/gpt-oss-120b\ (120B,\ A100 \times 1)$ | mxfp4 | 127 | 212 |
> | $openai/gpt-oss-120b\ (120B,\ MI300 \times 1)$ | mxfp4 | 191 | 311 |
> | $microsoft/VibeVoice-1.5B (A100 \times 1)$ | bf16 | - | rtf=0.07 |
> | $microsoft/VibeVoice-1.5B (MI300 \times 1)$ | bf16 | - | rtf=0.06 |
>

## What's New:

> Image-*20260808*: Support 1M context for Kimi K3 on MI300 192GB PCIe-5.
>
> Image-*20260707*: Memory fixes for 1M context on A100 80GB SXM.
>
> Image-*20260618*: Fit GLM-5.2 1M context into A100 80GB x 8 SXM.
>
> Image-*20260603*: Improved Claude Code Tooling Performance for GLM-5/5.1.
>
> Image-*20260511*: Initial Claude Code Support for GLM-5/5.1.
>
> Image-*20260327*: Add support for Kimi-K2.5.
>
> Image-*20260306*: Support DeepSeek V3.2 Long-context mode for A100/H100/MI300/B200.
>
> Image-*20251222*: Fine-tune A100 performance for most models.
>
> Image-*20251111*: Integrate Tutel LLM module into VibeVoice for accelerated inference (rtf = 0.07 for single A100).
>
> Image-*20251006*: Resolve compatibility with DeepSeek-V3.2-Exp
>
> Image-*20250827*: Add distributed support for OpenAI GPT-OSS 20B/120B with MXFP4 inference
>
> Image-*20250801*: Support Qwen3 MoE series, integrate [OpenWebUI](https://github.com/open-webui/open-webui)
>
> Image-*20250712*: Support Kimi K2 1TB MoE inference with NVFP4 for NVIDIA/AMD GPUs
>
> Image-*20250601*: Improved decoding performance for DeepSeek 671B on MI300x to 140-150 TPS
>
> More image versions can be found [here](https://hub.docker.com/r/tutelgroup/deepseek-671b/tags)

> Tutel v0.4.2: Add R1-FP4/Qwen3MoE-FP8 Support for NVIDIA and AMD GPUs & Fast Gating APIs:
```py
  >> Example:

import torch
from tutel import ops

# Qwen3 Fast MoE Gating for 128 Experts, with Routed Weights normalized to 1.0
logits_fp32 = torch.softmax(torch.randn([32, 128]), -1, dtype=torch.float32).cuda()
topk_weights, topk_ids = ops.qwen3_moe_scaled_topk(logits_fp32)
print(topk_weights, topk_ids, topk_weights.sum(-1))

# DeepSeek V3/R1 Fast MoE Gating for 256 Experts, with Routed Weights normalized to 2.5
logits_bf16 = torch.randn([32, 256], dtype=torch.bfloat16).cuda()
correction_bias_bf16 = torch.randn([logits_bf16.size(-1)], dtype=torch.bfloat16).cuda()
topk_weights, topk_ids = ops.deepseek_moe_sigmoid_scaled_topk(logits_bf16, correction_bias_bf16, None, None)
print(topk_weights, topk_ids, topk_weights.sum(-1))
```

> Tutel v0.4.1: Support Deepseek R1 FP8 with NVIDIA GPUs (A100 / A800)

> Tutel v0.4.0: Accelerating Deepseek R1 Full-precision-Chat for AMD MI300x8:
```sh
  >> Example:

# Step-1: Download Deepseek R1 671B Model
    huggingface-cli download deepseek-ai/DeepSeek-R1 --local-dir ./deepseek-ai/DeepSeek-R1

# Step-2: Using 8 MI300 GPUs to Serve Deepseek R1 Chat on Local Port :8000
    docker run -it --rm --ipc=host --privileged -p 8000:8000 \
        -v /:/host -w /host$(pwd) tutelgroup/deepseek-671b:mi300x8-chat-20250224 \
        --model_path ./deepseek-ai/DeepSeek-R1

# Step-3: Issue a Prompt Request with curl
    curl -X POST http://0.0.0.0:8000/chat -d '{"text": "Calculate the result of: 1 / (sqrt(5) - sqrt(3))"}'
```

> Tutel v0.3.3: Add all-to-all benchmark:
```sh
  >> Example:

python3 -m torch.distributed.run --nproc_per_node=8 -m tutel.examples.bandwidth_test --size_mb=256
```

> Tutel v0.3.2: Add tensorcore option for extra benchmarks / Extend the example for custom experts / Allow NCCL timeout settings:
```sh
  >> Example of using tensorcore:

python3 -m tutel.examples.helloworld --dtype=float32
    python3 -m tutel.examples.helloworld --dtype=float32 --use_tensorcore

python3 -m tutel.examples.helloworld --dtype=float16
    python3 -m tutel.examples.helloworld --dtype=float16 --use_tensorcore

>> Example of custom gates/experts:
    python3 -m tutel.examples.helloworld_custom_gate_expert --batch_size=16

>> Example of NCCL timeout settings:
    TUTEL_GLOBAL_TIMEOUT_SEC=60 python3 -m torch.distributed.run --nproc_per_node=8 -m tutel.examples.helloworld --use_tensorcore

```

> Tutel v0.3.1: Add NCCL all_to_all_v and all_gather_v for arbitrary-length message transfers:
```sh
  >> Example:
    # All_to_All_v:
    python3 -m torch.distributed.run --nproc_per_node=2 --master_port=7340 -m tutel.examples.nccl_all_to_all_v
    # All_Gather_v:
    python3 -m torch.distributed.run --nproc_per_node=2 --master_port=7340 -m tutel.examples.nccl_all_gather_v

>> How to:
    net.batch_all_to_all_v([t_x_cuda, t_y_cuda, ..], common_send_counts)
    net.batch_all_gather_v([t_x_cuda, t_y_cuda, ..])
```

> Tutel v0.3: Add Megablocks solution to improve decoder inference on single-GPU with num_local_expert >= 2:
```sh
  >> Example (capacity_factor=0 required by dropless-MoE):
    # Using BatchMatmul:
    python3 -m tutel.examples.helloworld --megablocks_size=0 --batch_size=1 --num_tokens=32 --top=1 --eval --num_local_experts=128 --capacity_factor=0
    # Using Megablocks with block_size = 1:
    python3 -m tutel.examples.helloworld --megablocks_size=1 --batch_size=1 --num_tokens=32 --top=1 --eval --num_local_experts=128 --capacity_factor=0
    # Using Megablocks with block_size = 2:
    python3 -m tutel.examples.helloworld --megablocks_size=2 --batch_size=1 --num_tokens=32 --top=1 --eval --num_local_experts=128 --capacity_factor=0

>> How to:
    self._moe_layer.forward(x, .., megablocks_size=1)         # Control the switch of megablocks_size (0 for disabled)
```

> Tutel v0.2: Allow most configurations to be dynamic switchable with free cost:
```sh
  >> Example:
    python3 -m torch.distributed.run --nproc_per_node=8 -m tutel.examples.helloworld_switch --batch_size=16

>> How to:
    self._moe_layer.forward(x, .., a2a_ffn_overlap_degree=2)  # Control the switch of overlap granularity (1 for no overlapping)
    self._moe_layer.forward(x, .., adaptive_r=1)              # Control the switch of parallelism (0 for DP, 1 for DP + EP, W / E for MP + EP, else for DP + MP + EP)
    self._moe_layer.forward(x, .., capacity_factor=1)         # Control the switch of capacity_volume (positive for padding, negative for no-padding, 0 for dropless)
    self._moe_layer.forward(x, .., top_k=1)                   # Control the switch of top_k sparsity
```

> Tutel v0.1: Optimize the Einsum Complexity of Data Dispatch Encoding and Decoding, add 2DH option to deal with All-to-All at scale:
```sh
  >> Example (suggest enabling 2DH only at scale, note that the value of --nproc_per_node MUST equal to total physical GPU counts per node, e.g. 8 for A100x8):
    python3 -m torch.distributed.run --nproc_per_node=8 -m tutel.examples.helloworld --batch_size=16 --use_2dh
```

-----------
## Getting Started

### 1. Prepare Pytorch (if applicable):
```
* Prepare Recommended Pytorch >= 2.0.0:
        #  Windows/Linux Pytorch for NVIDIA CUDA >= 11.7:
        python3 -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
        #  Linux Pytorch for AMD ROCm >= 6.2.2:
        python3 -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2.2
        #  Windows/Linux Pytorch for CPU:
        python3 -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
```

### 2. Tutel Installation:
```
* Option-1: Install Tutel Online:

$ python3 -m pip uninstall tutel -y
        $ python3 -m pip install -v -U --no-build-isolation git+https://github.com/microsoft/tutel@main

* Option-2: Build Tutel from Source:

$ git clone https://github.com/microsoft/tutel --branch main
        $ python3 -m pip uninstall tutel -y
        $ python3 ./tutel/setup.py install --user
```

### 3. Quick Test for Single Device / CPU:
```
* Quick Test on Single-GPU:

$ python3 -m tutel.examples.helloworld --batch_size=16               # Test Tutel-optimized MoE + manual distribution
        $ python3 -m tutel.examples.helloworld_ddp --batch_size=16           # Test Tutel-optimized MoE + Pytorch DDP distribution (requires: Pytorch >= 1.8.0)
        $ python3 -m tutel.examples.helloworld_ddp_tutel --batch_size=16     # Test Tutel-optimized MoE + Tutel DDP distribution (ZeRO on optimizors)
        $ python3 -m tutel.examples.helloworld_amp --batch_size=16           # Test Tutel-optimized MoE with AMP data type + manual distribution
        $ python3 -m tutel.examples.helloworld_custom_gate_expert --batch_size=16 # Test Tutel-optimized MoE + custom defined gate/expert layer
        $ python3 -m tutel.examples.helloworld_from_scratch                  # Test Custom MoE implementation from scratch
        $ python3 -m tutel.examples.moe_mnist                                # Test MoE layer in end-to-end MNIST dataset
        $ python3 -m tutel.examples.moe_cifar10                              # Test MoE layer in end-to-end CIFAR10 dataset

(If building from source, the following method also works:)
        $ python3 ./tutel/examples/helloworld.py --batch_size=16
        ..
```

### 4. Quick Test for 8 GPUs within 1 Machine:
```
        $ python3 -m torch.distributed.run --nproc_per_node=8 -m tutel.examples.helloworld --batch_size=16
```

### 5. Quick Test for Multiple GPUs across Machines:
```
* Run Tutel MoE in Distributed Mode:

(Option A - Torch launcher for `Multi-Node x Multi-GPU`:)
        $ ssh <node-ip-0> python3 -m torch.distributed.run --nproc_per_node=8 --nnodes=2 --node_rank=0 --master_addr=<node-ip-0> -m tutel.examples.helloworld --batch_size=16
        $ ssh <node-ip-1> python3 -m torch.distributed.run --nproc_per_node=8 --nnodes=2 --node_rank=1 --master_addr=<node-ip-0> -m tutel.examples.helloworld --batch_size=16

(Option B - Tutel launcher for `Multi-Node x Multi-GPU`, requiring package `openmpi-bin`:)
        # << Single Node >>
        $ mpiexec -bind-to none -host localhost -x LOCAL_SIZE=8 python3 -m tutel.launcher.run -m tutel.examples.helloworld_ddp_tutel --batch_size=16
        $ mpiexec -bind-to none -host localhost -x LOCAL_SIZE=8 python3 -m tutel.launcher.run -m tutel.examples.moe_mnist
        $ mpiexec -bind-to none -host localhost -x LOCAL_SIZE=8 python3 -m tutel.launcher.run -m tutel.examples.moe_cifar10
        ...

# << MPI-based launch for GPU backend>>
        $ mpiexec -bind-to none -host <node-ip-0>,<node-ip-1>,.. -x MASTER_ADDR=<node-ip-0> -x LOCAL_SIZE=8 python3 -m tutel.launcher.run -m tutel.examples.helloworld --batch_size=16

# << MPI-based Launch for CPU backend>>
        $ mpiexec -bind-to none -host localhost -x LOCAL_SIZE=1 -x OMP_NUM_THREADS=1024 python3 -m tutel.launcher.run -m tutel.examples.helloworld --batch_size=16 --device cpu
```

-----------

### Advance: Convert Checkpoint Files for Different World Sizes:
Documentation for checkpoint conversion has been moved [here](doc/CHECKPOINT.md).

### Examples: How to import Tutel-optimized MoE in Pytorch:
```
# Input Example:
import torch
x = torch.ones([6, 1024], device='cuda:0')

# Create MoE:
from tutel import moe as tutel_moe
moe_layer = tutel_moe.moe_layer(
    gate_type={'type': 'top', 'k': 2},
    model_dim=x.shape[-1],
    experts={
        'num_experts_per_device': 2,
        'type': 'ffn', 'hidden_size_per_expert': 2048, 'activation_fn': lambda x: torch.nn.functional.relu(x)
    },
    scan_expert_func = lambda name, param: setattr(param, 'skip_allreduce', True),
)

# Cast to GPU
moe_layer = moe_layer.to('cuda:0')

# In distributed model, you need further skip doing allreduce on global parameters that have `skip_allreduce` mask, 
# e.g.
#    for p in moe_layer.parameters():
#        if hasattr(p, 'skip_allreduce'):
#            continue
#        dist.all_reduce(p.grad)

# Forward MoE:
y = moe_layer(x)

print(y)
```

### Reference
You can consult this [paper](https://arxiv.org/pdf/2206.03382.pdf) below to get to know more technical details about Tutel:
```
@article {tutel,
author = {Changho Hwang and Wei Cui and Yifan Xiong and Ziyue Yang and Ze Liu and Han Hu and Zilong Wang and Rafael Salas and Jithin Jose and Prabhat Ram and Joe Chau and Peng Cheng and Fan Yang and Mao Yang and Yongqiang Xiong},
title = {Tutel: Adaptive Mixture-of-Experts at Scale},
year = {2022},
month = jun,
journal = {CoRR},
volume= {abs/2206.03382},
url = {https://arxiv.org/pdf/2206.03382.pdf},
}
```

### Usage of MOELayer:
```
* Usage of MOELayer Args:

gate_type        : dict-type gate description, e.g. {'type': 'top', 'k': 2, 'capacity_factor': -1.5, ..},
                              or a list of dict-type gate descriptions, e.g. [{'type': 'top', 'k', 2}, {'type': 'top', 'k', 2}],
                              the value of k in top-gating can be also negative, like -2, which indicates one GPU will hold 1/(-k) parameters of an expert
                              capacity_factor X can be positive (factor = X), zero (factor = max(needed_volumes)) or negative (factor = min(-X, max(needed_volumes))).
        model_dim        : the number of channels for MOE's input tensor
        experts          : a dict-type config for builtin expert network
        scan_expert_func : allow users to specify a lambda function to iterate each experts param, e.g. `scan_expert_func = lambda name, param: setattr(param, 'expert', True)`
        result_func      : allow users to specify a lambda function to format the MoE output and aux_loss, e.g. `result_func = lambda output: (output, output.l_aux)`
        group            : specify the explicit communication group of all_to_all
        seeds            : a tuple containing a tripple of int to specify manual seed of (shared params, local params, others params after MoE's)
        a2a_ffn_overlap_degree : the value to control a2a overlap depth, 1 by default for no overlap, 2 for overlap a2a with half gemm, ..
        parallel_type    : the parallel method to compute MoE, valid types: 'auto', 'data', 'model'
        pad_samples      : whether do auto padding on newly-coming input data to maximum data size in history

* Usage of dict-type Experts Config:

num_experts_per_device : the number of local experts per device (by default, the value is 1 if not specified)
        hidden_size_per_expert : the hidden size between two linear layers for each expert (used for type == 'ffn' only)
        type             : available built-in experts implementation, e.g: ffn
        activation_fn    : the custom-defined activation function between two linear layers (used for type == 'ffn' only)
        has_fc1_bias     : If set to False, the expert bias parameters `batched_fc1_bias` is disabled. Default: True
        has_fc2_bias     : If set to False, the expert bias parameters `batched_fc2_bias` is disabled. Default: True
```

### Contributing

This project welcomes contributions and suggestions.  Most contributions require you to agree to a
Contributor License Agreement (CLA) declaring that you have the right to, and actually do, grant us
the rights to use your contribution. For details, visit https://cla.opensource.microsoft.com.

When you submit a pull request, a CLA bot will automatically determine whether you need to provide
a CLA and decorate the PR appropriately (e.g., status check, comment). Simply follow the instructions
provided by the bot. You will only need to do this once across all repos using our CLA.

This project has adopted the [Microsoft Open Source Code of Conduct](https://opensource.microsoft.com/codeofconduct/).
For more information see the [Code of Conduct FAQ](https://opensource.microsoft.com/codeofconduct/faq/) or
contact [opencode@microsoft.com](mailto:opencode@microsoft.com) with any additional questions or comments.

### Trademarks

This project may contain trademarks or logos for projects, products, or services. Authorized use of Microsoft 
trademarks or logos is subject to and must follow 
[Microsoft's Trademark & Brand Guidelines](https://www.microsoft.com/en-us/legal/intellectualproperty/trademarks/usage/general).
Use of Microsoft trademarks or logos in modified versions of this project must not cause confusion or imply Microsoft sponsorship.
Any use of third-party trademarks or logos are subject to those third-party's policies.

Tutel

File Explorer

Today's picks

Was this content helpful?

Recent Feedback

Tutel

File Explorer

Today's picks

Was this content helpful?

Recent Feedback

Download README