MoE kernels AGPLv3

unsloth/kernels/moe/README.md

@@ -1,6 +1,7 @@
 ## MoE Grouped GEMM
 
 Optimized implementation of `MoE MLP Block`.
+Licensed under AGPLv3.
 
 ### Background
 

unsloth/kernels/moe/grouped_gemm/interface.py

@@ -1,3 +1,6 @@
+# SPDX-License-Identifier: GNU Affero General Public License v3.0
+# Copyright 2023-present the Unsloth team. All rights reserved.
+
 import logging
 import warnings
 from dataclasses import asdict

unsloth/kernels/moe/grouped_gemm/kernels/autotuning.py

@@ -1,3 +1,6 @@
+# SPDX-License-Identifier: GNU Affero General Public License v3.0
+# Copyright 2023-present the Unsloth team. All rights reserved.
+
 """
 Autotuning utils
 """

unsloth/kernels/moe/grouped_gemm/kernels/backward.py

@@ -1,3 +1,6 @@
+# SPDX-License-Identifier: GNU Affero General Public License v3.0
+# Copyright 2023-present the Unsloth team. All rights reserved.
+
 import torch
 import triton
 import triton.language as tl

unsloth/kernels/moe/grouped_gemm/kernels/forward.py

@@ -1,3 +1,6 @@
+# SPDX-License-Identifier: GNU Affero General Public License v3.0
+# Copyright 2023-present the Unsloth team. All rights reserved.
+
 import torch
 import triton
 import triton.language as tl

unsloth/kernels/moe/grouped_gemm/kernels/tuning.py

@@ -1,3 +1,6 @@
+# SPDX-License-Identifier: GNU Affero General Public License v3.0
+# Copyright 2023-present the Unsloth team. All rights reserved.
+
 """
 Manual tuning utils
 """

unsloth/kernels/moe/grouped_gemm/reference/layers/llama4_moe.py

@@ -1,3 +1,6 @@
+# SPDX-License-Identifier: GNU Affero General Public License v3.0
+# Copyright 2023-present the Unsloth team. All rights reserved.
+
 from dataclasses import dataclass
 from typing import Tuple
 

unsloth/kernels/moe/grouped_gemm/reference/layers/qwen3_moe.py

@@ -1,3 +1,6 @@
+# SPDX-License-Identifier: GNU Affero General Public License v3.0
+# Copyright 2023-present the Unsloth team. All rights reserved.
+
 from dataclasses import dataclass
 from typing import Tuple
 

unsloth/kernels/moe/grouped_gemm/reference/moe_block.py

@@ -1,3 +1,6 @@
+# SPDX-License-Identifier: GNU Affero General Public License v3.0
+# Copyright 2023-present the Unsloth team. All rights reserved.
+
 import torch
 from transformers.models.qwen3_moe.configuration_qwen3_moe import Qwen3MoeConfig
 from transformers.models.qwen3_moe.modeling_qwen3_moe import Qwen3MoeSparseMoeBlock

unsloth/kernels/moe/grouped_gemm/reference/moe_ops.py

@@ -1,3 +1,5 @@
+# SPDX-License-Identifier: GNU Affero General Public License v3.0
+# Copyright 2023-present the Unsloth team. All rights reserved.
 
 import torch
 import torch.nn.functional as F

unsloth/kernels/moe/tests/common.py

@@ -1,3 +1,6 @@
+# SPDX-License-Identifier: GNU Affero General Public License v3.0
+# Copyright 2023-present the Unsloth team. All rights reserved.
+
 import itertools
 from contextlib import contextmanager
 from dataclasses import dataclass, field

unsloth/kernels/moe/tests/moe_utils.py

@@ -1,3 +1,6 @@
+# SPDX-License-Identifier: GNU Affero General Public License v3.0
+# Copyright 2023-present the Unsloth team. All rights reserved.
+
 from dataclasses import dataclass, fields
 
 import torch

unsloth/kernels/moe/tests/test_grouped_gemm.py

@@ -1,3 +1,6 @@
+# SPDX-License-Identifier: GNU Affero General Public License v3.0
+# Copyright 2023-present the Unsloth team. All rights reserved.
+
 from dataclasses import asdict
 
 import pytest

unsloth/kernels/moe/tests/test_llama4_moe.py

@@ -1,3 +1,6 @@
+# SPDX-License-Identifier: GNU Affero General Public License v3.0
+# Copyright 2023-present the Unsloth team. All rights reserved.
+
 import argparse
 import sys
 from contextlib import contextmanager

unsloth/kernels/moe/tests/test_qwen3_moe.py

@@ -1,3 +1,6 @@
+# SPDX-License-Identifier: GNU Affero General Public License v3.0
+# Copyright 2023-present the Unsloth team. All rights reserved.
+
 import argparse
 from contextlib import contextmanager