Removing old doc. (#427)

* Removing old doc.

* Update pyo3 to get clippy happy.

* Optional MLX install.

* Mlx only on python 3.10.

.github/workflows/python.yml

@@ -58,6 +58,12 @@ jobs:
           pip install .[jax]
         shell: bash
 
+      - name: Install (mlx)
+        if: matrix.os == 'macos-latest' && matrix.version.python == "3.10"
+        run: |
+          pip install .[mlx]
+        shell: bash
+
       - name: Check style
         run: |
           pip install .[quality]

bindings/python/Cargo.toml

@@ -9,7 +9,7 @@ name = "safetensors_rust"
 crate-type = ["cdylib"]
 
 [dependencies]
-pyo3 = { version = "0.19.2", features = ["extension-module"] }
+pyo3 = { version = "0.20.2", features = ["extension-module"] }
 memmap2 = "0.5"
 serde_json = "1.0"
 

bindings/python/py_src/safetensors/flax.py

@@ -106,9 +106,6 @@ def load_file(filename: Union[str, os.PathLike]) -> Dict[str, Array]:
     Args:
         filename (`str`, or `os.PathLike`)):
             The name of the file which contains the tensors
-        device (`Dict[str, any]`, *optional*, defaults to `cpu`):
-            The device where the tensors need to be located after load.
-            available options are all regular flax device locations
 
     Returns:
         `Dict[str, Array]`: dictionary that contains name as key, value as `Array`

bindings/python/py_src/safetensors/numpy.py

@@ -111,9 +111,6 @@ def load_file(filename: Union[str, os.PathLike]) -> Dict[str, np.ndarray]:
     Args:
         filename (`str`, or `os.PathLike`)):
             The name of the file which contains the tensors
-        device (`Dict[str, any]`, *optional*, defaults to `cpu`):
-            The device where the tensors need to be located after load.
-            available options are all regular numpy device locations
 
     Returns:
         `Dict[str, np.ndarray]`: dictionary that contains name as key, value as `np.ndarray`

bindings/python/py_src/safetensors/tensorflow.py

@@ -105,9 +105,6 @@ def load_file(filename: Union[str, os.PathLike]) -> Dict[str, tf.Tensor]:
     Args:
         filename (`str`, or `os.PathLike`)):
             The name of the file which contains the tensors
-        device (`Dict[str, any]`, *optional*, defaults to `cpu`):
-            The device where the tensors need to be located after load.
-            available options are all regular tensorflow device locations
 
     Returns:
         `Dict[str, tf.Tensor]`: dictionary that contains name as key, value as `tf.Tensor`

bindings/python/pyproject.toml

@@ -50,6 +50,9 @@ jax = [
     "jax>=0.3.25",
     "jaxlib>=0.3.25",
 ]
+mlx = [
+    "mlx>=0.0.9",
+]
 paddlepaddle = [
     "safetensors[numpy]",
     "paddlepaddle>=2.4.1",