Add byte-progress reporting for safetensors file loading

CMakeLists.txt

@@ -30,7 +30,7 @@ endif()
 FetchContent_Declare(
   mlx-c
   GIT_REPOSITORY "https://github.com/ml-explore/mlx-c.git"
-  GIT_TAG "c74db5307cc8ce122f48d97ef951b30578674e7f")
+  GIT_TAG "2f8700a5797a4321597507ea57b3c54ffe4373af")
 FetchContent_MakeAvailable(mlx-c)
 
 # swift-numerics

Source/Cmlx/include-framework/mlx-c-io_types.h

@@ -37,10 +37,10 @@ typedef struct mlx_io_vtable_ {
   bool (*is_open)(void*);
   bool (*good)(void*);
   size_t (*tell)(void*);
-  void (*seek)(void*, int64_t off, int whence);
-  void (*read)(void*, char* data, size_t n);
-  void (*read_at_offset)(void*, char* data, size_t n, size_t off);
-  void (*write)(void*, const char* data, size_t n);
+  int (*seek)(void*, int64_t off, int whence);
+  size_t (*read)(void*, char* data, size_t n);
+  size_t (*read_at_offset)(void*, char* data, size_t n, size_t off);
+  size_t (*write)(void*, const char* data, size_t n);
   const char* (*label)(void*);
   void (*free)(void*);
 } mlx_io_vtable;

Source/Cmlx/include/mlx/c/io_types.h

@@ -37,10 +37,10 @@ typedef struct mlx_io_vtable_ {
   bool (*is_open)(void*);
   bool (*good)(void*);
   size_t (*tell)(void*);
-  void (*seek)(void*, int64_t off, int whence);
-  void (*read)(void*, char* data, size_t n);
-  void (*read_at_offset)(void*, char* data, size_t n, size_t off);
-  void (*write)(void*, const char* data, size_t n);
+  int (*seek)(void*, int64_t off, int whence);
+  size_t (*read)(void*, char* data, size_t n);
+  size_t (*read_at_offset)(void*, char* data, size_t n, size_t off);
+  size_t (*write)(void*, const char* data, size_t n);
   const char* (*label)(void*);
   void (*free)(void*);
 } mlx_io_vtable;

Source/MLX/Documentation.docc/MLX.md

@@ -130,3 +130,7 @@ See <doc:wired-memory> for configuration, best practices, and policy guidance.
 - ``Device``
 - ``DeviceType``
 - ``Stream``
+
+### Load Progress
+
+- ``LoadProgress``

Source/MLX/Documentation.docc/free-functions.md

@@ -120,9 +120,13 @@ operations as methods for convenience.
 
 - ``loadArray(url:stream:)``
 - ``loadArrays(url:stream:)``
+- ``loadArrays(url:stream:progressHandler:)``
 - ``loadArraysAndMetadata(url:stream:)``
+- ``loadArraysAndMetadata(url:stream:progressHandler:)``
 - ``save(array:url:stream:)``
 - ``save(arrays:metadata:url:stream:)``
+- ``withLoadProgressHandler(_:_:)-3ghip``
+- ``withLoadProgressHandler(_:_:)-8cm31``
 
 ### Logical