GWFAST on GPU

[Felipe-4]

Can I make GWFast run on the GPU, since it is written in Jax?

I was running the joint Fisher matrix of Ligo H1 and L1 for IMRPhenomD, frequency values between 20 and 1024 Hz (8033 elements in the asd array), with all 11 parameters. It took 4.4 seconds on a MacBook Air M1 8 GB of Ram. I want to see if I can speed that up, I just using the tutorial.

(Side question: from the paper is not clear to me, maybe I missed, is GWFast the fastest Fisher Matrix code compared to GWBENCH and GWFISH?)