updating JAX_FFT cuFFTMP to work with JAX 0.4.30

[ASKabalan]

Updating JAX_FFT code to work with latest JAX

[kvoronin]

Hi @ASKabalan

We have changed the license of the repository from BSD-3 to Apache 2.0 to allow accepting external contributions in the right way.

What this means is that

• the new default branch is main. (The old one called master is still there, but master and an extra bsd3_main are not supposed to be changed)
• all commits to the repository must be signed, see the new https://github.com/NVIDIA/CUDALibrarySamples/blob/main/CONTRIBUTING.md. Technically, it is as simple as doing git commit -s ... which would enforce DCO for the contribution (see more details in the CONTRIBUTING.md).

What this means for this PR:
Please change the target branch to main (potentially instead of rebase it is easier to cherry-pick commits into a new branch) and sign the commits.

Thanks!