subnets

Reusable network components.

Classes

MLP(*args, **kwargs)	Implements a flexible multi-layer perceptron (MLP) with optional residual connections, dropout, and spectral normalization.
ResidualUViT(*args, **kwargs)	Residual U-ViT backbone (SiD2-style [1]) for diffusion models.
TimeMLP(*args, **kwargs)	Implements a time-conditioned multi-layer perceptron (MLP).
UNet(*args, **kwargs)	Time-conditioned U-Net backbone for diffusion models [1].
UViT(*args, **kwargs)	Time-conditioned U-ViT backbone for diffusion models.