exposed scale_diffusion_across to python