GitHub repository: umangjpatel/kerax
Primary language: Python
Description: Keras-like APIs for JAX framework