diff --git a/compile.go b/internal/metal/compile.go
similarity index 99%
rename from compile.go
rename to internal/metal/compile.go
index 8440f4b..c1edeb3 100644
--- a/compile.go
+++ b/internal/metal/compile.go
@@ -1,6 +1,6 @@
 //go:build darwin && arm64
 
-package mlx
+package metal
 
 /*
 #include "mlx/c/mlx.h"
diff --git a/fast.go b/internal/metal/fast.go
similarity index 99%
rename from fast.go
rename to internal/metal/fast.go
index e1abeba..c2140fd 100644
--- a/fast.go
+++ b/internal/metal/fast.go
@@ -1,6 +1,6 @@
 //go:build darwin && arm64
 
-package mlx
+package metal
 
 /*
 #include <stdlib.h>
diff --git a/ops.go b/internal/metal/ops.go
similarity index 99%
rename from ops.go
rename to internal/metal/ops.go
index 99f953b..e027985 100644
--- a/ops.go
+++ b/internal/metal/ops.go
@@ -1,6 +1,6 @@
 //go:build darwin && arm64
 
-package mlx
+package metal
 
 /*
 #include <stdlib.h>
diff --git a/random.go b/internal/metal/random.go
similarity index 98%
rename from random.go
rename to internal/metal/random.go
index f7e09b2..bd6e90b 100644
--- a/random.go
+++ b/internal/metal/random.go
@@ -1,6 +1,6 @@
 //go:build darwin && arm64
 
-package mlx
+package metal
 
 /*
 #include "mlx/c/mlx.h"
diff --git a/slice.go b/internal/metal/slice.go
similarity index 99%
rename from slice.go
rename to internal/metal/slice.go
index 5bb7a66..494435f 100644
--- a/slice.go
+++ b/internal/metal/slice.go
@@ -1,6 +1,6 @@
 //go:build darwin && arm64
 
-package mlx
+package metal
 
 /*
 #include "mlx/c/mlx.h"