jimm docs
These are the docs for the Jax Image Modeling of Models library. These models are implemented using flax nnx/jax with sharding annotations for each model.
Models Implemented
- Vision Transformers
- CLIP
- more tbd
- contribute, it's open source!
These are the docs for the Jax Image Modeling of Models library. These models are implemented using flax nnx/jax with sharding annotations for each model.