GPU support (plans/roadmap)

modularml / mojo

The Mojo Programming Language

https://docs.modular.com/mojo/manual/

Other

22.96k stars 2.59k forks source link

GPU support (plans/roadmap) #3567

Open asmith26 opened 5 hours ago

asmith26 commented 5 hours ago

Hi there,

I'm thinking about buying a new GPU soon for neural network training/fine-tuning, and was wondering if mojo has plans to support (or currently supports) the various options:

NVIDIA (including e.g. any particular architectures/generations, CUDA versions/CUDA capability)
AMD (including e.g. any particular architectures, ROCm versions)
Intel (including e.g. any particular architectures/generations, oneAPI)

Many thanks for any help! :)

gryznar commented 3 hours ago

As far as I know, support for Nvidia GPU is coming with next Max release