you could also use "mps" (Apple), "xpu" (Intel), "hip" (AMD) GPUs and "xla" for TPUs. But yeah, those are less mature.
Alternative GPU and TPU Backends: MPS, XPU, HIP, XLA Options
By
–
By
–
you could also use "mps" (Apple), "xpu" (Intel), "hip" (AMD) GPUs and "xla" for TPUs. But yeah, those are less mature.