dzur658/mlx-unet-mnist

A Custom U-Net Designed and Trained in MLX

This is a toy model I made trained on the MNIST numbers dataset using MLX. It can draw the digits 0-9.

Github

Github Repo