This message was deleted.

hey @marah othman! We haven’t tested the library on non-nvidia gpus, but pytorch and pytorch-lightning support AMD gpus so it should work :)