I wrote a small Python library `mlxx` that monkey patches MLX, so we can conveniently use more array methods like `array.allclose(another)` or `vec1.inner(vec2)` without boilerplates.
Checkout github.com/ifsheldon/mlxx
Cheers!
Checkout github.com/ifsheldon/mlxx
Cheers!