azula.nn

Neural networks, layers and modules.

Submodules

azula.nn.attention	Attention layers.
azula.nn.dit	Diffusion Transformer (DiT) building blocks.
azula.nn.layers	Common layers.
azula.nn.unet	U-Net building blocks.
azula.nn.utils	Miscellaneous neural network helpers.
azula.nn.vit	Vision Transformer (ViT) building blocks.