added bitwise inversion operations, extension and freestanding (#288)

Author: AppAppWorks

Source/MLX/Documentation.docc/Organization/arithmetic.md

@@ -98,6 +98,7 @@ Note: the `-` and `/` operators are not able to be linked here.
 - ``MLXArray/.>=(_:_:)-6zxj9``
 - ``MLXArray/.&&(_:_:)``
 - ``MLXArray/.||(_:_:)``
+- ``MLXArray/~(_:)``
 - ``MLXArray/&(_:_:)-9in7a``
 - ``MLXArray/&(_:_:)-8js41``
 - ``MLXArray/&(_:_:)-7iu4h``
@@ -146,6 +147,7 @@ Note: the `-` and `/` operators are not able to be linked here.
 - ``atan2(_:_:stream:)``
 - ``atanh(_:stream:)``
 - ``bitwiseAnd(_:_:stream:)``
+- ``bitwiseInvert(_:stream:)``
 - ``bitwiseOr(_:_:stream:)``
 - ``bitwiseXOr(_:_:stream:)``
 - ``ceil(_:stream:)``

Source/MLX/MLXArray+Ops.swift

@@ -681,6 +681,20 @@ extension MLXArray {
         return MLXArray(result)
     }
 
+    /// Unary element-wise bitwise invert.
+    ///
+    /// Apply bitwise inversion to the values in the array.
+    ///
+    /// ### See Also
+    /// - <doc:arithmetic>
+    /// - ``bitwiseInvert(_:stream:)``
+    public static prefix func ~ (lhs: MLXArray) -> MLXArray {
+        let s = StreamOrDevice.default
+        var result = mlx_array_new()
+        mlx_bitwise_invert(&result, lhs.ctx, s.ctx)
+        return MLXArray(result)
+    }
+
     /// Element-wise bitwise and.
     ///
     /// Take the bitwise and of two arrays with numpy-style broadcasting

Source/MLX/Ops+Array.swift

@@ -358,6 +358,18 @@ public func arrayEqual<T: ScalarOrArray>(
     return MLXArray(result)
 }
 
+/// Unary element-wise bitwise invert.
+///
+/// Apply bitwise inversion to the values in the array.
+///
+/// ### See Also
+/// - <doc:arithmetic>
+public func bitwiseInvert(_ array: MLXArray, stream: StreamOrDevice = .default) -> MLXArray {
+    var result = mlx_array_new()
+    mlx_bitwise_invert(&result, array.ctx, stream.ctx)
+    return MLXArray(result)
+}
+
 /// Element-wise bitwise and.
 ///
 /// Take the bitwise and of two arrays with numpy-style broadcasting