All the code was done in TensorFlow [1]} and is open sourcedhttps://github.com/zaccharieramzi/fastmri-reproducible-benchmark.
The Non-uniform Fast Fourier Transform in particular is implemented as a TensorFlow packagehttps://github.com/zaccharieramzi/tfkbnufft inspired by the original PyTorch implementation from [2]}.
