[nbdev]
diffusion_curvature = diffusion_curvature._modidx:d
