mlx: a mix between numpy and cuda/hip for metal