deprecate pytorch backend in favor of JAX

[flaport]

JAX might potentially offer considerable speedups and offers better overlap with the numpy api.

[gomezzz]

Just stumbled on this, you can have both fairly easily with https://github.com/jcmgray/autoray :)

Also allows for a numpy-like API , see e.g. here https://github.com/FHof/torchquad/blob/develop/torchquad/integration/trapezoid.py

[flaport]

Thanks @gomezzz ,

This looks pretty cool indeed! I'll look into it 🙂