Skip to content

Fix GM gpu timings

Philipp Frank requested to merge fix_gm_gpu_timings into NIFTy_8

Replaces the generic jax_fori loop with cumsums for the WP and IWP

Merge request reports