5 repos
Artur-Galstyan/jaxonmodels
JAX models for ML
12
201 commits
jax-ml/bonsai
Minimal, lightweight JAX implementations of popular models.
242
179 commits
clementpoiret/equimo
Implementation of Deep Learning Models in JAX (equinox)
25
240 commits
RhizomeResearch/Equimo
pythoncrazy/jimm
JAX Image Modeling of Models contains Computer Vision/Vision Language Model implementations in…
7
165 commits
