Equimo 0.3.3__tar.gz → 0.3.4__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (39) hide show
  1. {equimo-0.3.3 → equimo-0.3.4}/Equimo.egg-info/PKG-INFO +3 -1
  2. equimo-0.3.4/Equimo.egg-info/SOURCES.txt +12 -0
  3. {equimo-0.3.3 → equimo-0.3.4}/PKG-INFO +3 -1
  4. equimo-0.3.4/equimo/__init__.py +1 -0
  5. {equimo-0.3.3 → equimo-0.3.4}/pyproject.toml +1 -2
  6. equimo-0.3.3/Equimo.egg-info/SOURCES.txt +0 -35
  7. equimo-0.3.3/equimo/__init__.py +0 -1
  8. equimo-0.3.3/equimo/layers/__init__.py +0 -0
  9. equimo-0.3.3/equimo/layers/activation.py +0 -33
  10. equimo-0.3.3/equimo/layers/attention.py +0 -1615
  11. equimo-0.3.3/equimo/layers/convolution.py +0 -329
  12. equimo-0.3.3/equimo/layers/downsample.py +0 -58
  13. equimo-0.3.3/equimo/layers/dropout.py +0 -128
  14. equimo-0.3.3/equimo/layers/ffn.py +0 -381
  15. equimo-0.3.3/equimo/layers/generic.py +0 -70
  16. equimo-0.3.3/equimo/layers/mamba.py +0 -169
  17. equimo-0.3.3/equimo/layers/norm.py +0 -165
  18. equimo-0.3.3/equimo/layers/patch.py +0 -401
  19. equimo-0.3.3/equimo/layers/posemb.py +0 -416
  20. equimo-0.3.3/equimo/layers/sharing.py +0 -160
  21. equimo-0.3.3/equimo/layers/squeeze_excite.py +0 -143
  22. equimo-0.3.3/equimo/models/__init__.py +0 -6
  23. equimo-0.3.3/equimo/models/emamodel.py +0 -29
  24. equimo-0.3.3/equimo/models/fastervit.py +0 -493
  25. equimo-0.3.3/equimo/models/mlla.py +0 -173
  26. equimo-0.3.3/equimo/models/partialformer.py +0 -411
  27. equimo-0.3.3/equimo/models/shvit.py +0 -381
  28. equimo-0.3.3/equimo/models/vit.py +0 -501
  29. equimo-0.3.3/equimo/models/vssd.py +0 -196
  30. equimo-0.3.3/equimo/ops/scan.py +0 -246
  31. {equimo-0.3.3 → equimo-0.3.4}/Equimo.egg-info/dependency_links.txt +0 -0
  32. {equimo-0.3.3 → equimo-0.3.4}/Equimo.egg-info/requires.txt +0 -0
  33. {equimo-0.3.3 → equimo-0.3.4}/Equimo.egg-info/top_level.txt +0 -0
  34. {equimo-0.3.3 → equimo-0.3.4}/LICENSE.md +0 -0
  35. {equimo-0.3.3 → equimo-0.3.4}/README.md +0 -0
  36. {equimo-0.3.3 → equimo-0.3.4}/equimo/io.py +0 -0
  37. {equimo-0.3.3 → equimo-0.3.4}/equimo/utils.py +0 -0
  38. {equimo-0.3.3 → equimo-0.3.4}/setup.cfg +0 -0
  39. {equimo-0.3.3 → equimo-0.3.4}/tests/test_models.py +0 -0
{equimo-0.3.3 → equimo-0.3.4}/Equimo.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: Equimo
3
- Version: 0.3.3
3
+ Version: 0.3.4
4
4
  Summary: Implementation of popular vision models in Jax
5
5
  Classifier: Development Status :: 4 - Beta
6
6
  Classifier: Intended Audience :: Developers
@@ -13,6 +13,7 @@ Classifier: Programming Language :: Python :: 3.13
13
13
  Classifier: Programming Language :: Python :: 3.14
14
14
  Requires-Python: >=3.11
15
15
  Description-Content-Type: text/markdown
16
+ License-File: LICENSE.md
16
17
  Requires-Dist: einops>=0.8.0
17
18
  Requires-Dist: equinox>=0.11.5
18
19
  Requires-Dist: jax>=0.4.25
@@ -31,6 +32,7 @@ Provides-Extra: extras
31
32
  Requires-Dist: matplotlib>=3.10.1; extra == "extras"
32
33
  Requires-Dist: pillow>=11.1.0; extra == "extras"
33
34
  Requires-Dist: scikit-learn>=1.6.1; extra == "extras"
35
+ Dynamic: license-file
34
36
 
35
37
  # Equimo: Modern Vision Models in JAX/Equinox
36
38
 
equimo-0.3.4/Equimo.egg-info/SOURCES.txt ADDED
@@ -0,0 +1,12 @@
1
+ LICENSE.md
2
+ README.md
3
+ pyproject.toml
4
+ Equimo.egg-info/PKG-INFO
5
+ Equimo.egg-info/SOURCES.txt
6
+ Equimo.egg-info/dependency_links.txt
7
+ Equimo.egg-info/requires.txt
8
+ Equimo.egg-info/top_level.txt
9
+ equimo/__init__.py
10
+ equimo/io.py
11
+ equimo/utils.py
12
+ tests/test_models.py
{equimo-0.3.3 → equimo-0.3.4}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: Equimo
3
- Version: 0.3.3
3
+ Version: 0.3.4
4
4
  Summary: Implementation of popular vision models in Jax
5
5
  Classifier: Development Status :: 4 - Beta
6
6
  Classifier: Intended Audience :: Developers
@@ -13,6 +13,7 @@ Classifier: Programming Language :: Python :: 3.13
13
13
  Classifier: Programming Language :: Python :: 3.14
14
14
  Requires-Python: >=3.11
15
15
  Description-Content-Type: text/markdown
16
+ License-File: LICENSE.md
16
17
  Requires-Dist: einops>=0.8.0
17
18
  Requires-Dist: equinox>=0.11.5
18
19
  Requires-Dist: jax>=0.4.25
@@ -31,6 +32,7 @@ Provides-Extra: extras
31
32
  Requires-Dist: matplotlib>=3.10.1; extra == "extras"
32
33
  Requires-Dist: pillow>=11.1.0; extra == "extras"
33
34
  Requires-Dist: scikit-learn>=1.6.1; extra == "extras"
35
+ Dynamic: license-file
34
36
 
35
37
  # Equimo: Modern Vision Models in JAX/Equinox
36
38
 
equimo-0.3.4/equimo/__init__.py ADDED
@@ -0,0 +1 @@
1
+ __version__ = "0.3.4"
{equimo-0.3.3 → equimo-0.3.4}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "Equimo"
3
- version = "0.3.3"
3
+ version = "0.3.4"
4
4
  description = "Implementation of popular vision models in Jax"
5
5
  readme = "README.md"
6
6
  requires-python = ">=3.11"
@@ -54,5 +54,4 @@ dev = [
54
54
  ]
55
55
 
56
56
  [tool.setuptools]
57
- license-files = []
58
57
  packages = ["equimo"]
equimo-0.3.3/Equimo.egg-info/SOURCES.txt DELETED
@@ -1,35 +0,0 @@
1
- LICENSE.md
2
- README.md
3
- pyproject.toml
4
- Equimo.egg-info/PKG-INFO
5
- Equimo.egg-info/SOURCES.txt
6
- Equimo.egg-info/dependency_links.txt
7
- Equimo.egg-info/requires.txt
8
- Equimo.egg-info/top_level.txt
9
- equimo/__init__.py
10
- equimo/io.py
11
- equimo/utils.py
12
- equimo/layers/__init__.py
13
- equimo/layers/activation.py
14
- equimo/layers/attention.py
15
- equimo/layers/convolution.py
16
- equimo/layers/downsample.py
17
- equimo/layers/dropout.py
18
- equimo/layers/ffn.py
19
- equimo/layers/generic.py
20
- equimo/layers/mamba.py
21
- equimo/layers/norm.py
22
- equimo/layers/patch.py
23
- equimo/layers/posemb.py
24
- equimo/layers/sharing.py
25
- equimo/layers/squeeze_excite.py
26
- equimo/models/__init__.py
27
- equimo/models/emamodel.py
28
- equimo/models/fastervit.py
29
- equimo/models/mlla.py
30
- equimo/models/partialformer.py
31
- equimo/models/shvit.py
32
- equimo/models/vit.py
33
- equimo/models/vssd.py
34
- equimo/ops/scan.py
35
- tests/test_models.py
equimo-0.3.3/equimo/__init__.py DELETED
@@ -1 +0,0 @@
1
- __version__ = "0.3.3"
equimo-0.3.3/equimo/layers/__init__.py DELETED
File without changes
equimo-0.3.3/equimo/layers/activation.py DELETED
@@ -1,33 +0,0 @@
1
- from typing import Callable
2
- import jax
3
- from functools import partial
4
-
5
-
6
- def get_act(activation: str | Callable) -> Callable:
7
- """Get an activation function from its common name.
8
-
9
- This is necessary because configs have to be stringified and stored as
10
- json files to allow (de)serialization.
11
- """
12
- if not isinstance(activation, str):
13
- return activation
14
-
15
- match activation:
16
- case "relu":
17
- return jax.nn.relu
18
- case "gelu":
19
- return jax.nn.gelu
20
- case "exactgelu":
21
- return partial(jax.nn.gelu, approximate=False)
22
- case "silu":
23
- return jax.nn.silu
24
- case "elu":
25
- return jax.nn.elu
26
- case "sigmoid":
27
- return jax.nn.sigmoid
28
- case "hard_sigmoid":
29
- return jax.nn.hard_sigmoid
30
- case "softmax":
31
- return jax.nn.softmax
32
- case _:
33
- raise ValueError(f"Got an unknown activation string: {activation}")