GitHub repository: leandrolcampos/minJAX
Primary language: Python
Description: A minimal and didactic implementation of the fundamental concepts in JAX's core system