Describe the solution you'd like
MLX seems to be adding also support for Nvidia GPUs , making it really interesting as could improve our support both on Mac and Linux boxes
Describe alternatives you've considered
Additional context
ml-explore/mlx#1983