apid experimentation and cleaner abstractions in the code, where the model prediction is separated from the sample update.
To this end, the design of schedulers is such that:
Schedulers can be used interchangeably between diffusion models in inference to find the preferred trade-off between speed and generation quality.
Schedulers are currently by default in PyTorch, but are designed to be framework independent (partial Jax support currently ex