TorchLean API

NN.Spec.Generative.Diffusion

Diffusion / flow specs (umbrella) #

This is the curated public entrypoint for TorchLean's diffusion / flow spec layer.

It re-exports:

All specs are scalar-polymorphic (Context α) so the same definitions can be reused for: