Does xformer require CUDA or can it also be run on TPU/Tranium instances?

[Tommaso Bendinelli]

❓ Questions and Help

The titles sums it all, do I need a CUDA device or is Xformers compatible with other devices?

[dan_the_3rd]

Hi,
xFormers supports only CUDA at the moment, but we have experimental (and limited) support for AMD. There is nothing planned for TPU or Trainium