Assist in creating MLX implementations of UNet, NestedUnet, etc

[luke-carlson]

No description provided.

[bdeanhardt]

Ethan and I can take this one!

[luke-carlson]

Hey how's this one going? Feel free to reach out with any questions

[bdeanhardt]

Hi Luke! I started an MLX_UNET.py and wrote the MLP class based on your example. I am going to put a PR up for that now. I haven’t written any tests for it yet, but to be honest I’m a little bit confused how I would test it in the first place. I’m still trying to figure that out. Thanks for checking in, Bella

…

On Dec 4, 2024, at 9:40 AM, Luke Carlson ***@***.***> wrote: Hey how's this one going? Feel free to reach out with any questions — Reply to this email directly, view it on GitHub <#45 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/BBJ6TPX2YXI4V63DY5E2ZDD2D4H65AVCNFSM6AAAAABSJOL6CCVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDKMJXGYZTAOBVGI>. You are receiving this because you commented.

[gabrielfnayres]

Hi @luke-carlson, I did the Unet implementation using mlx and wrote a test to compare it with torch.

#61