search

apple / ml-mdm

Train high-quality text-to-image diffusion models in a data & compute efficient manner
https://machinelearning.apple.com/research/matryoshka-diffusion-models
MIT License
448 stars 31 forks source link

Assist in creating MLX implementations of UNet, NestedUnet, etc #45

Open luke-carlson opened 3 days ago

bdeanhardt commented 2 days ago

Ethan and I can take this one!