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.
- 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
- Install Jax: Follow the installation instructions for Jax
- Clone the Repository: Clone this repository to your local machine using
git clone https://github.com/noahzhy/UNet_Lite_Jax.git. - To be continued