Bump extension MLX version (#1350)

* Bump extension MLX version

* fix some docs nits

Author: awni

docs/src/python/ops.rst

@@ -44,6 +44,7 @@ Operations
    convolve
    conv1d
    conv2d
+   conv3d
    conv_general
    cos
    cosh

examples/extensions/pyproject.toml

@@ -2,7 +2,7 @@
 requires = [
   "setuptools>=42",
   "cmake>=3.24",
-  "mlx>=0.16.2",
+  "mlx>=0.17.0",
   "nanobind==2.1.0",
 ]
 build-backend = "setuptools.build_meta"

examples/extensions/requirements.txt

@@ -1,4 +1,4 @@
 setuptools>=42
 cmake>=3.24
-mlx>=0.16.2
+mlx>=0.17.0
 nanobind==2.1.0

python/src/fast.cpp

@@ -102,11 +102,12 @@ void init_fast(nb::module_& parent_module) {
               implementation which rotates consecutive dimensions.
             base (float, optional): The base used to compute angular frequency for
               each dimension in the positional encodings. Exactly one of ``base`` and
-             ``freqs`` must be ``None``.
+              ``freqs`` must be ``None``.
             scale (float): The scale used to scale the positions.
             offset (int): The position offset to start at.
             freqs (array, optional): Optional frequencies to use with RoPE.
-              If set, the ``base`` parameter must be ``None``. ``Default: None``.
+              If set, the ``base`` parameter must be ``None``. Default: ``None``.
+
         Returns:
             array: The output array.
       )pbdoc");