Equimo

View on PyPIReverse Dependencies (0)

0.3.3 equimo-0.3.3-py3-none-any.whl

Wheel Details

Project: Equimo
Version: 0.3.3
Filename: equimo-0.3.3-py3-none-any.whl
Download: [link]
Size: 63719
MD5: b738ee9b10f4fe50102642cbf9e08f6b
SHA256: 91877aa04bd8b37c74527776d62ff92f9935309f0640429f0289acdc2b6f3261
Uploaded: 2025-03-26 15:57:47 +0000

dist-info

METADATA

Metadata-Version: 2.4
Name: Equimo
Version: 0.3.3
Summary: Implementation of popular vision models in Jax
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: MIT License
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Programming Language :: Python :: 3.14
Requires-Python: >=3.11
Requires-Dist: einops (>=0.8.0)
Requires-Dist: equinox (>=0.11.5)
Requires-Dist: jax (>=0.4.25)
Requires-Dist: jaxlib (>=0.4.25)
Requires-Dist: loguru (>=0.7.3)
Requires-Dist: lz4 (>=4.4.3)
Requires-Dist: requests (>=2.32.3)
Requires-Dist: semver (>=3.0.4)
Requires-Dist: torch (>=2.6.0); extra == "torch"
Requires-Dist: timm (>=1.0.15); extra == "torch"
Requires-Dist: tensorflow (>=2.18.1); extra == "text"
Requires-Dist: tensorflow-text (>=2.18.1); extra == "text"
Requires-Dist: matplotlib (>=3.10.1); extra == "extras"
Requires-Dist: pillow (>=11.1.0); extra == "extras"
Requires-Dist: scikit-learn (>=1.6.1); extra == "extras"
Provides-Extra: torch
Provides-Extra: text
Provides-Extra: extras
Description-Content-Type: text/markdown
[Description omitted; length: 8115 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (78.1.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
equimo/__init__.py sha256=8KcCYTXH99C2-gCLuPILJvtT9YftRWJsartIx6TQ2ZY 22
equimo/io.py sha256=1zSRyezgRmZ4DDjzCunIjgUu6OUsO4eibaurAtq5P3M 9626
equimo/utils.py sha256=TyMvsNpoP-QYodZT3kQpaeaqWULq_Bz5bAvQ2OaZR9s 5909
equimo/layers/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
equimo/layers/activation.py sha256=o-S2A9n-ZKflGL8y3Avb5u6fYkekWJZex9xf608xTtE 967
equimo/layers/attention.py sha256=ZsBAHgMv8I5f7500Q7c8CPUR0w-lLCM5ZzvQFcYsDAs 49286
equimo/layers/convolution.py sha256=HV4tQc1hdC9Oc2psPgrXY1Xdbub9G-yHRw4KQgWzu9s 10532
equimo/layers/downsample.py sha256=z3l6iiww8-k976q-WlbYk1iaSeyniMg7RD_Q-BwZV-Y 1691
equimo/layers/dropout.py sha256=H83oQKUhfjLPcVZ9rKWjvxIBNj7kOzHQgOWkErdgTOA 4218
equimo/layers/ffn.py sha256=4kzJMFj1E2r3UvvIAn-2xRIiQr25pyK6t3zwmANnRXI 11810
equimo/layers/generic.py sha256=Qju1cay_ub2JGnLODtTPbtw3C2M3-NPAiIEtHM_N-UU 2181
equimo/layers/mamba.py sha256=OKUnTaKXP-eRN-wK6cD8mzRgf1bqOmAcj5XkArV6IhA 5589
equimo/layers/norm.py sha256=VDvRaqbVXekDXF_Uh2mOJxZ0LYeQwsot_jWenLzKOL0 4324
equimo/layers/patch.py sha256=YNLXcBWWK9BXYAxZI1FjV9O88OVihp23FLtpwNT1618 12715
equimo/layers/posemb.py sha256=eAa7yc4MxeGJGadApJrFnEYh4h275eF53YxV4_QM0Go 13392
equimo/layers/sharing.py sha256=h0YKh9obCzM2yv1_onAN3ieAZpqoZ7UsHq1areJf7GY 3996
equimo/layers/squeeze_excite.py sha256=BVhuX2s_sQStdKd-Xg3sn76d7r1AW8JmRlVHtN7LTtM 4142
equimo/models/__init__.py sha256=9EmYrBZe5datt2_WlbjXLdGEgvCvwrDpKMTaYNLBz1Q 180
equimo/models/emamodel.py sha256=YctJzbgr-s4RoBexSQMPdghQYw5_NaJEdS_opM_G72k 755
equimo/models/fastervit.py sha256=xuXOK5o7kky9fwkj1bnDff_iI23IKOXS4j5JO0CzYWs 15163
equimo/models/mlla.py sha256=fZgtLl-mcFlXJUgJy1mRJ9V0hqkl6JtSXRBqCqkc_xQ 5702
equimo/models/partialformer.py sha256=sJBxrI__iF-SM8GtTobjUw6dAG0EBCsvcUvoY6TMHYg 12911
equimo/models/shvit.py sha256=5tG3HiHTHuTX2AjeLuMGpnUwTlfISp57AcpY4aIuIjY 11300
equimo/models/vit.py sha256=dv0jLLU3qTyCdJ8YoChNjJlyVBZMMRuzkAkkdlomC80 16973
equimo/models/vssd.py sha256=__BiJolHk618P9Q5Jyk0mEo4l9RM_D7rglAfvc_h-38 6467
equimo/ops/scan.py sha256=1v80gkIkmQ0QoUj5YJpVQQ7ZjF3AfPfQchGLqSOqf4g 8729
equimo-0.3.3.dist-info/METADATA sha256=N-1GxGhGv2Qg704dTuqilpYKikUto-pox1lTJho00ok 9416
equimo-0.3.3.dist-info/WHEEL sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0 91
equimo-0.3.3.dist-info/top_level.txt sha256=wfqgodqy_2pxpHWYoY89Fzr2I0-zPOER8zQ5bHWUxH0 7
equimo-0.3.3.dist-info/RECORD

top_level.txt

equimo