Peter Tsoi

[Feature] cross product

1

It would be nice to have a `mx.cross`/`mx.linalg.cross` that works like `np.cross`/`np.linalg.cross`.

[Feature] Metal inverse (`mx.linalg.inv`)

9

**Describe the bug** When trying to invert a small 3x3 (camera intrinsics matrix), mlx crashes. **To Reproduce** ``` import mlx.core as mx intrinsics = mx.array([[1162.38, 0, 618.25], [0, 1156.83, 410.75],...

[Feature] custom extensions

5

Currently, if I want to write a new custom operation and use it in Swift, I would have to maintain my own fork of MLX. It would be great to...

Can we bring the `custom_vjp` goodness recently exposed in Python (ml-explore/mlx#1246) to Swift?