UNet-Lite with Jax

This repository provides an implementation of UNet-Lite, a lightweight variant of the popular UNet architecture for image segmentation, using the high-performance numerical computation library Jax for Python.

Key Features

• Efficient implementation of UNet-Lite using Jax's vectorized operations and functional programming paradigm
• Modular code structure for easy customization and experimentation
• Support for training and evaluation on image segmentation datasets

Getting Started

1. Install Jax: Follow the installation instructions for Jax
2. Clone the Repository: Clone this repository to your local machine using git clone https://github.com/noahzhy/UNet_Lite_Jax.git.
3. To be continued