X Tutup

Name		Name	Last commit message	Last commit date
parent directory ..
README.md		README.md
__init__.py		__init__.py
attention.py		attention.py
attention_flax.py		attention_flax.py
embeddings.py		embeddings.py
embeddings_flax.py		embeddings_flax.py
resnet.py		resnet.py
resnet_flax.py		resnet_flax.py
unet_2d.py		unet_2d.py
unet_2d_condition.py		unet_2d_condition.py
unet_2d_condition_flax.py		unet_2d_condition_flax.py
unet_blocks.py		unet_blocks.py
unet_blocks_flax.py		unet_blocks_flax.py
vae.py		vae.py
vae_flax.py		vae_flax.py

README.md

Models

Models: Neural network that models $p_\theta(\mathbf{x}_{t-1}|\mathbf{x}_t)$ (see image below) and is trained end-to-end to denoise a noisy input to an image. Examples: UNet, Conditioned UNet, 3D UNet, Transformer UNet

API

TODO(Suraj, Patrick)

Examples

TODO(Suraj, Patrick)

X Tutup