GitHub repository: poets-ai/elegy
Primary language: Python
Description: A High Level API for Deep Learning in JAX