My understanding is that once JAX takes off, the cuda advantage is gone for nvidia. That's a big if/when though.