Reproduction of the grokking paper using jax.
Primary LanguagePythonMIT LicenseMIT
No issues in this repository yet.