/sjax

Simple JAX pseudo-framework built on top of Haiku's tutorial.

Primary LanguageJupyter Notebook

Stargazers