From bbef2e8adc32fa0d580df0f95d6ea242b63b0923 Mon Sep 17 00:00:00 2001
From: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com>
Date: Thu, 6 Nov 2025 09:19:47 +0100
Subject: [PATCH] add kernel

---
 src/transformers/integrations/hub_kernels.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/transformers/integrations/hub_kernels.py b/src/transformers/integrations/hub_kernels.py
index 112ac670e9a0..6d9512d15a3d 100644
--- a/src/transformers/integrations/hub_kernels.py
+++ b/src/transformers/integrations/hub_kernels.py
@@ -71,6 +71,12 @@
                     layer_name="RMSNorm",
                 )
             },
+            "mps": {
+                Mode.INFERENCE: LayerRepository(
+                    repo_id="kernels-community/mlx_rmsnorm",
+                    layer_name="RMSNorm",
+                )
+            },
         },
         "MLP": {
             "cuda": LayerRepository(