├── .envrc ├── .github └── workflows │ └── python-tests.yaml ├── .gitignore ├── .python-version ├── LICENSE.md ├── README.md ├── devenv.lock ├── devenv.nix ├── devenv.yaml ├── docs └── assets │ ├── 0-shot.png │ └── demo.jpg ├── models ├── dinov2.py ├── dinov3.py ├── siglip2.py ├── tips.py └── tips_text.py ├── pyproject.toml ├── src └── equimo │ ├── __init__.py │ ├── conversion │ ├── __init__.py │ └── utils.py │ ├── experimental │ ├── __init__.py │ └── text.py │ ├── io.py │ ├── layers │ ├── __init__.py │ ├── activation.py │ ├── attention.py │ ├── convolution.py │ ├── downsample.py │ ├── dropout.py │ ├── ffn.py │ ├── generic.py │ ├── mamba.py │ ├── norm.py │ ├── patch.py │ ├── posemb.py │ ├── sharing.py │ ├── squeeze_excite.py │ └── wavelet.py │ ├── models │ ├── __init__.py │ ├── attnet.py │ ├── emamodel.py │ ├── fastervit.py │ ├── lowformer.py │ ├── mlla.py │ ├── partialformer.py │ ├── reduceformer.py │ ├── shvit.py │ ├── vit.py │ └── vssd.py │ ├── ops │ └── scan.py │ └── utils.py ├── tests └── test_models.py └── uv.lock /.envrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/.envrc -------------------------------------------------------------------------------- /.github/workflows/python-tests.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/.github/workflows/python-tests.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/.gitignore -------------------------------------------------------------------------------- /.python-version: -------------------------------------------------------------------------------- 1 | 3.12 2 | -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/README.md -------------------------------------------------------------------------------- /devenv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/devenv.lock -------------------------------------------------------------------------------- /devenv.nix: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/devenv.nix -------------------------------------------------------------------------------- /devenv.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/devenv.yaml -------------------------------------------------------------------------------- /docs/assets/0-shot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/docs/assets/0-shot.png -------------------------------------------------------------------------------- /docs/assets/demo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/docs/assets/demo.jpg -------------------------------------------------------------------------------- /models/dinov2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/models/dinov2.py -------------------------------------------------------------------------------- /models/dinov3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/models/dinov3.py -------------------------------------------------------------------------------- /models/siglip2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/models/siglip2.py -------------------------------------------------------------------------------- /models/tips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/models/tips.py -------------------------------------------------------------------------------- /models/tips_text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/models/tips_text.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/pyproject.toml -------------------------------------------------------------------------------- /src/equimo/__init__.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.5.0-alpha.4" 2 | -------------------------------------------------------------------------------- /src/equimo/conversion/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /src/equimo/conversion/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/conversion/utils.py -------------------------------------------------------------------------------- /src/equimo/experimental/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /src/equimo/experimental/text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/experimental/text.py -------------------------------------------------------------------------------- /src/equimo/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/io.py -------------------------------------------------------------------------------- /src/equimo/layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/equimo/layers/activation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/layers/activation.py -------------------------------------------------------------------------------- /src/equimo/layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/layers/attention.py -------------------------------------------------------------------------------- /src/equimo/layers/convolution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/layers/convolution.py -------------------------------------------------------------------------------- /src/equimo/layers/downsample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/layers/downsample.py -------------------------------------------------------------------------------- /src/equimo/layers/dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/layers/dropout.py -------------------------------------------------------------------------------- /src/equimo/layers/ffn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/layers/ffn.py -------------------------------------------------------------------------------- /src/equimo/layers/generic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/layers/generic.py -------------------------------------------------------------------------------- /src/equimo/layers/mamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/layers/mamba.py -------------------------------------------------------------------------------- /src/equimo/layers/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/layers/norm.py -------------------------------------------------------------------------------- /src/equimo/layers/patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/layers/patch.py -------------------------------------------------------------------------------- /src/equimo/layers/posemb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/layers/posemb.py -------------------------------------------------------------------------------- /src/equimo/layers/sharing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/layers/sharing.py -------------------------------------------------------------------------------- /src/equimo/layers/squeeze_excite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/layers/squeeze_excite.py -------------------------------------------------------------------------------- /src/equimo/layers/wavelet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/layers/wavelet.py -------------------------------------------------------------------------------- /src/equimo/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/models/__init__.py -------------------------------------------------------------------------------- /src/equimo/models/attnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/models/attnet.py -------------------------------------------------------------------------------- /src/equimo/models/emamodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/models/emamodel.py -------------------------------------------------------------------------------- /src/equimo/models/fastervit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/models/fastervit.py -------------------------------------------------------------------------------- /src/equimo/models/lowformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/models/lowformer.py -------------------------------------------------------------------------------- /src/equimo/models/mlla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/models/mlla.py -------------------------------------------------------------------------------- /src/equimo/models/partialformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/models/partialformer.py -------------------------------------------------------------------------------- /src/equimo/models/reduceformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/models/reduceformer.py -------------------------------------------------------------------------------- /src/equimo/models/shvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/models/shvit.py -------------------------------------------------------------------------------- /src/equimo/models/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/models/vit.py -------------------------------------------------------------------------------- /src/equimo/models/vssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/models/vssd.py -------------------------------------------------------------------------------- /src/equimo/ops/scan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/ops/scan.py -------------------------------------------------------------------------------- /src/equimo/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/src/equimo/utils.py -------------------------------------------------------------------------------- /tests/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/tests/test_models.py -------------------------------------------------------------------------------- /uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementpoiret/Equimo/HEAD/uv.lock --------------------------------------------------------------------------------