diff --git a/mlx/ops.h b/mlx/ops.h
index 50f29afa1..7046f7a39 100644
--- a/mlx/ops.h
+++ b/mlx/ops.h
@@ -2,6 +2,7 @@
 
 #pragma once
 
+#include <optional>
 #include <variant>
 
 #include "array.h"