/jax-deep-models

A collection of deep learning models written in JAX/Flax.

Primary LanguageJupyter Notebook

Watchers