Add MPS backend for Apple Silicon

README.md

@@ -158,10 +158,10 @@ bitsandbytes has the following minimum requirements for all platforms:
     <tr>
       <td></td>
       <td>⬜ Metal <br><code>mps</code></td>
-      <td>Apple M1+</td>
-      <td>🐢</td>
-      <td>🐢</td>
-      <td>❌</td>
+      <td>Apple M1+<br><small>Install <a href="https://github.com/mpsops/mps-bitsandbytes">mps-bitsandbytes</a> for optimized Metal kernels</small></td>
+      <td>✅</td>
+      <td>✅</td>
+      <td>✅</td>
   </tbody>
 </table>
 

bitsandbytes/__init__.py

@@ -45,6 +45,13 @@
     if hasattr(torch, "hpu") and torch.hpu.is_available():
         from .backends.hpu import ops as hpu_ops
 
+# MPS backend (Apple Silicon) - requires mps-bitsandbytes for optimized Metal kernels
+if hasattr(torch.backends, "mps") and torch.backends.mps.is_available():
+    try:
+        from .backends.mps import ops as mps_ops
+    except ImportError:
+        pass  # mps-bitsandbytes not installed, will use default backend
+
 
 def _import_backends():
     """

bitsandbytes/backends/mps/__init__.py

@@ -0,0 +1,4 @@
+# MPS backend for bitsandbytes
+# Provides optimized Metal kernels for Apple Silicon
+
+from . import ops