GitHub repository: Kaushalya/jax_playground
Primary language: Python
Description: Pure-functional neural networks in JAX