Pure JAX implementation of the Non-Uniform Fast Fourier Transform (NUFFT)
Python 29 3
JAX/GPU N4 bias field correction — a fast drop-in match for SimpleITK N4 (~2600x faster on A100, <0.2% match)
Python 4 1