I can't find any docs that numpy will do this.

https://opensource.apple.com/projects/mlx/ is needed to do this?