Skip to content

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
  • SigLIP
  • more tbd
  • contribute, it's open source!

Installation

Using pixi.sh:

pixi add jimm@https://github.com/Locamage/jimm.git --pypi

Using uv

uv add --dev git+https://github.com/Locamage/jimm.git or if you prefer to not add as a direct dependency: uv pip install git+https://github.com/Locamage/jimm.git

Using pip/conda

pip install git+https://github.com/Locamage.git