ml-mdm icon indicating copy to clipboard operation
ml-mdm copied to clipboard

Assist in creating MLX implementations of UNet, NestedUnet, etc

Open luke-carlson opened this issue 1 year ago • 3 comments

luke-carlson avatar Nov 22 '24 14:11 luke-carlson

Ethan and I can take this one!

bdeanhardt avatar Nov 23 '24 18:11 bdeanhardt

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

luke-carlson avatar Dec 04 '24 14:12 luke-carlson

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 https://github.com/apple/ml-mdm/issues/45#issuecomment-2517630852, or unsubscribe https://github.com/notifications/unsubscribe-auth/BBJ6TPX2YXI4V63DY5E2ZDD2D4H65AVCNFSM6AAAAABSJOL6CCVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDKMJXGYZTAOBVGI. You are receiving this because you commented.

bdeanhardt avatar Dec 04 '24 15:12 bdeanhardt