search

NVIDIA / CUDALibrarySamples

CUDA Library Samples
Other
1.5k stars 311 forks source link

updating JAX_FFT cuFFTMP to work with JAX 0.4.30 #198

Open ASKabalan opened 1 month ago

ASKabalan commented 1 month ago

Updating JAX_FFT code to work with latest JAX