Add a mx.metal.device_info (#1060)

* device inof

* add variant

* fix linux

* fix doc

python/src/metal.cpp

@@ -4,6 +4,8 @@
 #include <nanobind/nanobind.h>
 #include <nanobind/stl/optional.h>
 #include <nanobind/stl/string.h>
+#include <nanobind/stl/unordered_map.h>
+#include <nanobind/stl/variant.h>
 
 namespace nb = nanobind;
 using namespace nb::literals;
@@ -116,4 +118,19 @@ void init_metal(nb::module_& m) {
       R"pbdoc(
       Stop a Metal capture.
       )pbdoc");
+  metal.def(
+      "device_info",
+      &metal::device_info,
+      R"pbdoc(
+      Get information about the GPU device and system settings.
+
+      Currently returns:
+
+      * ``architecture``
+      * ``max_buffer_size``
+      * ``max_recommended_working_set_size``
+
+      Returns:
+          dict: A dictionary with string keys and string or integer values.
+      )pbdoc");
 }