subnets

Reusable network components.

Classes

MLP(*args, **kwargs)	Multi-layer perceptron built from DenseBlock layers.
ResidualUViT(*args, **kwargs)	Residual U-ViT backbone (SiD2-style [1]) for diffusion models.
TimeMLP(*args, **kwargs)	Time-conditioned multi-layer perceptron with FiLM modulation.
UNet(*args, **kwargs)	Time-conditioned U-Net backbone for diffusion models [1].
UViT(*args, **kwargs)	Time-conditioned U-ViT backbone for diffusion models.