diff --git a/python/src/linalg.cpp b/python/src/linalg.cpp
index 7bd186d51..c193060db 100644
--- a/python/src/linalg.cpp
+++ b/python/src/linalg.cpp
@@ -1,22 +1,13 @@
-
 // Copyright © 2023 Apple Inc.
 
-#include <limits>
-#include <numeric>
-#include <ostream>
-#include <string>
 #include <variant>
 
-#include <pybind11/iostream.h>
 #include <pybind11/pybind11.h>
 #include <pybind11/stl.h>
 
 #include "mlx/linalg.h"
-#include "mlx/ops.h"
-#include "mlx/utils.h"
 
 #include "python/src/load.h"
-#include "python/src/overloaded.h"
 #include "python/src/utils.h"
 
 namespace py = pybind11;
diff --git a/python/src/overloaded.h b/python/src/overloaded.h
deleted file mode 100644
index 204a3178a..000000000
--- a/python/src/overloaded.h
+++ /dev/null
@@ -1,8 +0,0 @@
-// Copyright © 2023 Apple Inc.
-
-template <typename... Ts>
-struct overloaded : Ts... {
-  using Ts::operator()...;
-};
-template <class... Ts>
-overloaded(Ts...) -> overloaded<Ts...>;
\ No newline at end of file