Can SimPEG use GPU?

Hi, I would like to know if there is a way to use GPU with SimPEG. Especially doing 3D inversions, which usually take a lot of time. Thanks

Hi @anagmantilla. Thanks for asking this question. Currently SimPEG doesn’t provide a way to run forward models or inversions in GPU. Since the underlying pieces are based on Numpy arrays, implementations of forward models that can run in GPU would require a lot of rewrite using PyTorch or similar libraries.

Nonetheless, I’m curious about what parts of the code take lot of time. Is it the forward modeling, or the inversion steps? What type of inversion are you running?