▪ U-Net-Segmentation-Jax

Jax-based implementation of U-net for image segemetation task using the Oxford-IIIT Pet Dataset.
[Blog Post]     [Code]