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