Equimo 0.4.1__tar.gz → 0.4.2__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 (44) hide show
  1. {equimo-0.4.1/src/Equimo.egg-info → equimo-0.4.2}/PKG-INFO +1 -1
  2. {equimo-0.4.1 → equimo-0.4.2}/pyproject.toml +1 -1
  3. {equimo-0.4.1 → equimo-0.4.2/src/Equimo.egg-info}/PKG-INFO +1 -1
  4. equimo-0.4.2/src/equimo/__init__.py +1 -0
  5. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/models/reduceformer.py +5 -4
  6. equimo-0.4.1/src/equimo/__init__.py +0 -1
  7. {equimo-0.4.1 → equimo-0.4.2}/LICENSE.md +0 -0
  8. {equimo-0.4.1 → equimo-0.4.2}/README.md +0 -0
  9. {equimo-0.4.1 → equimo-0.4.2}/setup.cfg +0 -0
  10. {equimo-0.4.1 → equimo-0.4.2}/src/Equimo.egg-info/SOURCES.txt +0 -0
  11. {equimo-0.4.1 → equimo-0.4.2}/src/Equimo.egg-info/dependency_links.txt +0 -0
  12. {equimo-0.4.1 → equimo-0.4.2}/src/Equimo.egg-info/requires.txt +0 -0
  13. {equimo-0.4.1 → equimo-0.4.2}/src/Equimo.egg-info/top_level.txt +0 -0
  14. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/conversion/__init__.py +0 -0
  15. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/conversion/utils.py +0 -0
  16. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/experimental/__init__.py +0 -0
  17. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/experimental/text.py +0 -0
  18. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/io.py +0 -0
  19. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/layers/__init__.py +0 -0
  20. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/layers/activation.py +0 -0
  21. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/layers/attention.py +0 -0
  22. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/layers/convolution.py +0 -0
  23. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/layers/downsample.py +0 -0
  24. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/layers/dropout.py +0 -0
  25. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/layers/ffn.py +0 -0
  26. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/layers/generic.py +0 -0
  27. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/layers/mamba.py +0 -0
  28. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/layers/norm.py +0 -0
  29. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/layers/patch.py +0 -0
  30. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/layers/posemb.py +0 -0
  31. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/layers/sharing.py +0 -0
  32. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/layers/squeeze_excite.py +0 -0
  33. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/models/__init__.py +0 -0
  34. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/models/emamodel.py +0 -0
  35. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/models/fastervit.py +0 -0
  36. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/models/lowformer.py +0 -0
  37. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/models/mlla.py +0 -0
  38. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/models/partialformer.py +0 -0
  39. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/models/shvit.py +0 -0
  40. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/models/vit.py +0 -0
  41. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/models/vssd.py +0 -0
  42. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/ops/scan.py +0 -0
  43. {equimo-0.4.1 → equimo-0.4.2}/src/equimo/utils.py +0 -0
  44. {equimo-0.4.1 → equimo-0.4.2}/tests/test_models.py +0 -0
{equimo-0.4.1/src/Equimo.egg-info → equimo-0.4.2}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: Equimo
3
- Version: 0.4.1
3
+ Version: 0.4.2
4
4
  Summary: Implementation of popular vision models in Jax
5
5
  Classifier: Development Status :: 4 - Beta
6
6
  Classifier: Intended Audience :: Developers
{equimo-0.4.1 → equimo-0.4.2}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "Equimo"
3
- version = "0.4.1"
3
+ version = "0.4.2"
4
4
  description = "Implementation of popular vision models in Jax"
5
5
  readme = "README.md"
6
6
  requires-python = ">=3.11"
{equimo-0.4.1 → equimo-0.4.2/src/Equimo.egg-info}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: Equimo
3
- Version: 0.4.1
3
+ Version: 0.4.2
4
4
  Summary: Implementation of popular vision models in Jax
5
5
  Classifier: Development Status :: 4 - Beta
6
6
  Classifier: Intended Audience :: Developers
equimo-0.4.2/src/equimo/__init__.py ADDED
@@ -0,0 +1 @@
1
+ __version__ = "0.4.2"
{equimo-0.4.1 → equimo-0.4.2}/src/equimo/models/reduceformer.py RENAMED
@@ -159,7 +159,7 @@ class ReduceFormer(eqx.Module):
159
159
  block_types: list[Literal["conv", "attention"]],
160
160
  *,
161
161
  key: PRNGKeyArray,
162
- heads_dim: int = 32,
162
+ head_dim: int = 32,
163
163
  expand_ratio: float = 4.0,
164
164
  norm_layer: eqx.Module | str = eqx.nn.GroupNorm,
165
165
  act_layer: Callable | str = jax.nn.hard_swish,
@@ -227,6 +227,7 @@ class ReduceFormer(eqx.Module):
227
227
  depth=depth,
228
228
  block_type=block_type,
229
229
  stride=2,
230
+ head_dim=head_dim,
230
231
  expand_ratio=expand_ratio,
231
232
  norm_layer=norm_layer,
232
233
  act_layer=act_layer,
@@ -335,7 +336,7 @@ def reduceformer_backbone_b1(**kwargs) -> ReduceFormer:
335
336
  "attention",
336
337
  "attention",
337
338
  ],
338
- heads_dim=16,
339
+ head_dim=16,
339
340
  **kwargs,
340
341
  )
341
342
  return backbone
@@ -352,7 +353,7 @@ def reduceformer_backbone_b2(**kwargs) -> ReduceFormer:
352
353
  "attention",
353
354
  "attention",
354
355
  ],
355
- heads_dim=32,
356
+ head_dim=32,
356
357
  **kwargs,
357
358
  )
358
359
  return backbone
@@ -369,7 +370,7 @@ def reduceformer_backbone_b3(**kwargs) -> ReduceFormer:
369
370
  "attention",
370
371
  "attention",
371
372
  ],
372
- heads_dim=32,
373
+ head_dim=32,
373
374
  **kwargs,
374
375
  )
375
376
  return backbone
equimo-0.4.1/src/equimo/__init__.py DELETED
@@ -1 +0,0 @@
1
- __version__ = "0.4.1"
{equimo-0.4.1 → equimo-0.4.2}/LICENSE.md RENAMED
File without changes
{equimo-0.4.1 → equimo-0.4.2}/README.md RENAMED
File without changes
{equimo-0.4.1 → equimo-0.4.2}/setup.cfg RENAMED
File without changes
{equimo-0.4.1 → equimo-0.4.2}/src/equimo/io.py RENAMED
File without changes
{equimo-0.4.1 → equimo-0.4.2}/src/equimo/layers/ffn.py RENAMED
File without changes
{equimo-0.4.1 → equimo-0.4.2}/src/equimo/layers/norm.py RENAMED
File without changes
{equimo-0.4.1 → equimo-0.4.2}/src/equimo/models/mlla.py RENAMED
File without changes
{equimo-0.4.1 → equimo-0.4.2}/src/equimo/models/vit.py RENAMED
File without changes
{equimo-0.4.1 → equimo-0.4.2}/src/equimo/models/vssd.py RENAMED
File without changes
{equimo-0.4.1 → equimo-0.4.2}/src/equimo/ops/scan.py RENAMED
File without changes
{equimo-0.4.1 → equimo-0.4.2}/src/equimo/utils.py RENAMED
File without changes
{equimo-0.4.1 → equimo-0.4.2}/tests/test_models.py RENAMED
File without changes