Also, it does not work with pytorch, which means it does not work with anything really.

Hopefully this will soon change.

Right now you can choose between cpu, nvidia or Apple neural engine.