ENH: tools for diffusion