jimmy-vision

View on PyPIReverse Dependencies (0)

0.0.7 jimmy_vision-0.0.7-py3-none-any.whl

Wheel Details

Project: jimmy-vision
Version: 0.0.7
Filename: jimmy_vision-0.0.7-py3-none-any.whl
Download: [link]
Size: 50796
MD5: cffd6fa789eee9d2f39850a32d9d8421
SHA256: 3020df1e422d0985d394dfe4da05aabbb51dee2f43a8fc2ed314b26973364011
Uploaded: 2024-08-27 13:43:05 +0000

dist-info

METADATA

Metadata-Version: 2.3
Name: jimmy-vision
Version: 0.0.7
Summary: Implementations of popular vision models in Jax (Flax NNX)
Requires-Python: >=3.11
Requires-Dist: einops (>=0.8.0)
Requires-Dist: flax (>=0.8.5)
Requires-Dist: jax (>=0.4.27)
Requires-Dist: py7zr (>=0.21.1)
Requires-Dist: urllib3 (>=2.0.0)
Requires-Dist: jax-cuda12-plugin[with-cuda] (>=0.4.27); extra == "cuda12"
Requires-Dist: jaxlib (>=0.4.27); extra == "cuda12"
Requires-Dist: torch (==2.4.0); extra == "torch"
Requires-Dist: torchvision (==0.19.0); extra == "torch"
Provides-Extra: cuda12
Provides-Extra: torch
Description-Content-Type: text/markdown
License-Expression: MIT
License-File: LICENSE.md
[Description omitted; length: 4611 characters]

WHEEL

Wheel-Version: 1.0
Generator: hatchling 1.25.0
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
jimmy/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jimmy/convert.py sha256=L1EqIARnCMvg8Wht7i61oopzlQId6uJmaCsbE1MV1FM 3496
jimmy/io.py sha256=rqbJf3qMcOhP4GMB36G-Af1s0GQFbE1xqiwZsEAe0BM 3798
jimmy/utils.py sha256=k6lfiA4V-W0vtwCLHPwIdbmuneNP89ELwSE5LSC52as 4017
jimmy/layers/__init__.py sha256=wMkgxtU_m_8Hr9S4RAMEmIQc6PFDcjxmYXI3rRaaf-I 540
jimmy/layers/attention.py sha256=56fPeemzCsXN85peK3GK0WvCrXHrtWWfHtuWT0GKnSk 10168
jimmy/layers/blocks.py sha256=cYNf2ZtPc3jDq0X-LfLBknmzehOK65kGcOChX7gjI9M 21814
jimmy/layers/builders.py sha256=DVI7vT06sjJu7vFl-fqdpXmKW5WjnpMWzjFIEBfmv2w 1429
jimmy/layers/configs.py sha256=dfxpGYn32IcDWYoTh2_3RS82hoiASo3gLequIKJKNzs 3972
jimmy/layers/dropout.py sha256=rJkVQlWWX02iEkFZyGW9eRdnJRk6a8uDSemfxrVXX5E 2771
jimmy/layers/fastervit.py sha256=y3VKbTQCB-f2AnC3LA21q70zAHjMxK_hGa6lbsDHa0E 6453
jimmy/layers/generic.py sha256=1xUW85PAEL_bZIAb8f0w3TysVZ8qOLkeut39yB1YXik 4318
jimmy/layers/mamba.py sha256=Jo4U1-bYuEPkWyxjUjItRu_GfT9VIhZTkK1uv8tX_-o 16869
jimmy/layers/misc.py sha256=21YF_kploF19bobQJdNsb3YzxHUrYBn5YUfQJmORxoI 1675
jimmy/layers/mlp.py sha256=1R1QsPI3eqKfVSY3KImc8MJ-yBuz4VwkAHsagjdciK4 2112
jimmy/layers/norm.py sha256=2tnxslsNzMV4XDXc-HAXv1CNkuneRlMXW9cRdnX5OWU 1543
jimmy/layers/patch.py sha256=rV5ugX_ggxRrMelgy10EeaSDu5dx7wGUPnrA2ZVfyLg 13255
jimmy/layers/posemb.py sha256=BDiRSYt5Hlp1Ovp255_fF_GWAaxlusrDG0XT3Gc1YWA 11585
jimmy/layers/sharing.py sha256=40mDMxj6orePguMVaggIM5qRqcc41hf5uN20TIt8Bcs 2818
jimmy/layers/swiglu.py sha256=Ye685z4ieicguDcwz7RE28aPCa6QVyIWikI2YMLq4ww 1800
jimmy/models/__init__.py sha256=tMjCBqidUNyYJr3-h9zfqNMkJi75urQCCeYqO6xG3Dw 54
jimmy/models/configs.py sha256=m4fni7Q8b6A5MRA0XsgDWOpDZZjykHFSMyno_6MByFM 2684
jimmy/models/emamodel.py sha256=TfPx7KUkuzZYCvZuT9ZKsJh2EXVazYoiqifmF3pgDRc 3170
jimmy/models/fastervit.py sha256=WuzktbzCBnQvhTvPd_vYixh5r_3v63K0_CyxJ-MppSE 6318
jimmy/models/loader.py sha256=0lrQ3MPK_sX5JFYqGpBz4DzJs-T9K3UE4i14gSXB2s8 2056
jimmy/models/mamba.py sha256=gj56VtaoOyU8Agg0OH51ByixdZaJHN7pFQ7p3T1ZEBc 7598
jimmy/models/mlla.py sha256=QzdrMmg_MpW1Gwqblx0OvSRvzYMgtTYm4eFGkqW1jSI 5900
jimmy/models/vit.py sha256=s6ihBJQLPAfLRgO4aUhfEYyncnb1yt5RMswFdHg-CuA 12235
jimmy/ops/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jimmy/ops/scan.py sha256=vze7sUk4YhjQuHNCzwAQQtze9bBq6y-xQPeb37hhEPg 8797
jimmy_vision-0.0.7.dist-info/METADATA sha256=Z7Cqm1M4hOjThPW_1O5cr6a4GSznc5qOiVx7Hk8a4lg 5256
jimmy_vision-0.0.7.dist-info/WHEEL sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY 87
jimmy_vision-0.0.7.dist-info/licenses/LICENSE.md sha256=NDoiESQBdng_tbHOUj6BvMHpxu5P8JtsRbqVz-4rNIM 1072
jimmy_vision-0.0.7.dist-info/RECORD