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

Published 20 hours ago • verified publisher icon apple

MLX_UNET.py

Open bdeanhardt opened this issue 1 year ago • 1 comments

a first pass at the MLX implementation for UNET

bdeanhardt avatar Dec 04 '24 15:12 bdeanhardt

Because the test case for MLP_MLX passes, you could start by just merging in that class implantation? If you'd like to merge others, we'd want to make similar test cases comparing them to the pytorch versions before merging

luke-carlson avatar Dec 11 '24 17:12 luke-carlson