GitHub repository: stickshift/xjax
Primary language: Jupyter Notebook
Description: Exploring Jax