.. py:module:: diffusion .. currentmodule:: diffusion ============= API Reference ============= .. toctree:: :maxdepth: 2 diffusion schedule