Equimo 2.8.0__tar.gz → 2.8.1__tar.gz

{equimo-2.8.0 → equimo-2.8.1}/CHANGELOG.md

@@ -2,6 +2,13 @@
 
 All notable changes to Equimo are documented here. Equimo follows Semantic Versioning from version 2.0.0 onward.
 
+## [2.8.1] - 2026-11-11
+
+### Fixed
+
+- DEQ models run in bfloat16. Equimo now requires banax 0.1.5, whose fixed-point solvers keep their convergence errors
+  in float32 instead of failing on a dtype change inside the solver loop.
+
 ## [2.8.0] - 2026-11-10
 
 ### Added

{equimo-2.8.0 → equimo-2.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Equimo
-Version: 2.8.0
+Version: 2.8.1
 Summary: Implementation of modern multimodal models in JAX/Equinox
 License-Expression: MIT AND Apache-2.0
 License-File: LICENSE.md
@@ -27,7 +27,7 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
-Requires-Dist: banax>=0.1.2
+Requires-Dist: banax>=0.1.5
 Requires-Dist: einops>=0.8.0
 Requires-Dist: equinox>=0.13.3
 Requires-Dist: jax>=0.11.1

equimo-2.8.0/pyproject.toml.orig → equimo-2.8.1/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "Equimo"
-version = "2.8.0"
+version = "2.8.1"
 description = "Implementation of modern multimodal models in JAX/Equinox"
 readme = "README.md"
 requires-python = ">=3.12,<3.15"
@@ -25,7 +25,7 @@ license-files = [
   "LICENSES/tfc-t0-APACHE-2.0.txt",
 ]
 dependencies = [
-  "banax>=0.1.2",
+  "banax>=0.1.5",
   "einops>=0.8.0",
   "equinox>=0.13.3",
   "jax>=0.11.1",
@@ -96,7 +96,7 @@ wheel-exclude = ["EGM.md"]
 
 [dependency-groups]
 dev = [
-    "jax2onnx>=0.17.0",
+    "jax2onnx>=0.19.0",
     "mdformat==1.0.0",
     "mdformat-footnote==0.1.3",
     "mdformat-gfm==1.0.0",

equimo-2.8.1/src/equimo/__init__.py

@@ -0,0 +1 @@
+__version__ = "2.8.1"

{equimo-2.8.0 → equimo-2.8.1}/src/equimo/core/implicit/EGM.md

@@ -116,9 +116,12 @@ Re-exported by `equimo.core.implicit` ([`__init__.py`](./__init__.py)): `DEQBloc
   `key=jr.PRNGKey(42)` is created once, at import time.
 - `[deq.py:DEQCell.__init__]`: the default `strategy=EntryInjection()` is a single instance created at import and shared
   by every caller that uses the default.
-- `[deq.py:DEQBlock]`: depends on the external `banax` package. There is no I/O, printing or host callback.
+- `[deq.py:DEQBlock]`: depends on the external `banax` package, at least 0.1.5, whose solvers keep their convergence
+  errors in float32, so a bfloat16 `DEQBlock` solves and compares `tol` in float32. There is no I/O, printing or host
+  callback.
 
 ## Verification
 
 - Run: `devenv shell -- uv run pytest tests/test_implicit.py -q`
-- `tests/test_implicit.py` has no low-precision case.
+- `tests/test_implicit.py` has no low-precision case. The `deq` model case runs `DEQ` in bfloat16 against float32:
+  `devenv shell -- uv run pytest tests/finetune/test_feature_spec_model_coverage.py -k deq -q`.

{equimo-2.8.0 → equimo-2.8.1}/src/equimo/vision/layers/EGM.md

@@ -201,7 +201,7 @@ shared core layers (activation, dropout, FFN, generic, Mamba mixer, norm) so mod
   `[posemb.py:VisionRoPE]`, `[posemb.py:PosEmbMLPSwinv2D]`
 - `[posemb.py:_static_table]` evaluates those tables even while tracing, so `jax.jit` programs and ONNX exports hold
   them as constants; `tests/test_onnx_export.py::test_posemb_tables_export_as_constants` checks that each table exports
-  as a constant. Exporting `RoPE` and the Swin tables needs a jax2onnx release that includes enpasos/jax2onnx#291.
+  as a constant. Exporting `RoPE` and the Swin tables needs jax2onnx 0.19.0 or newer (enpasos/jax2onnx#291).
   `LinearAttention` applies its `RoPE` to queries and keys in one call, so programs hold one table per block.
   `[attention.py:LinearAttention]`
 - `LearnedPosEmbed` validates `cls_token` and `reg_tokens` shapes and resamples the grid part only, leaving prefix rows
@@ -226,4 +226,4 @@ shared core layers (activation, dropout, FFN, generic, Mamba mixer, norm) so mod
   `devenv shell -- uv run pytest tests/test_attention.py tests/test_convolution.py tests/test_downsample.py tests/test_patch.py tests/test_posemb.py tests/test_squeeze_excite.py tests/test_wavelet.py tests/test_layer_registry_coverage.py tests/test_generic.py -q`
 - The computed position-embedding tables are also covered by
   `devenv shell -- uv run pytest tests/test_onnx_export.py::test_posemb_tables_export_as_constants tests/test_io.py tests/test_checkpoint_signature_stability.py -q`
-  (constant export, which needs jax2onnx with enpasos/jax2onnx#291, and loading pre-2.8 checkpoints).
+  (constant export, which needs jax2onnx 0.19.0 or newer, and loading pre-2.8 checkpoints).

equimo-2.8.0/pyproject.toml

@@ -1,136 +0,0 @@
-[project]
-name = "Equimo"
-version = "2.8.0"
-description = "Implementation of modern multimodal models in JAX/Equinox"
-readme = "README.md"
-requires-python = ">=3.12,<3.15"
-license = "MIT AND Apache-2.0"
-license-files = [
-    "LICENSE.md",
-    "NOTICE",
-    "LICENSES/Apache-2.0.txt",
-    "LICENSES/pretrained/README.md",
-    "LICENSES/pretrained/ast-BSD-3-Clause.txt",
-    "LICENSES/pretrained/convnext-Apache-2.0.txt",
-    "LICENSES/pretrained/convnextv2-CC-BY-NC-4.0.txt",
-    "LICENSES/pretrained/dinov2-Apache-2.0.txt",
-    "LICENSES/pretrained/dinov3-License.md",
-    "LICENSES/pretrained/eupe-FAIR-Noncommercial-Research-License.md",
-    "LICENSES/pretrained/lingbot-vision-Apache-2.0.txt",
-    "LICENSES/pretrained/sam-License.md",
-    "LICENSES/pretrained/siglip2-Apache-2.0.txt",
-    "LICENSES/pretrained/t0-alpha-Apache-2.0.txt",
-    "LICENSES/pretrained/tabpfn-3-License-v1.0.txt",
-    "LICENSES/pretrained/tips-CC-BY-4.0.txt",
-    "LICENSES/tfc-t0-APACHE-2.0.txt",
-]
-dependencies = [
-    "banax>=0.1.2",
-    "einops>=0.8.0",
-    "equinox>=0.13.3",
-    "jax>=0.11.1",
-    "jaxlib>=0.11.1",
-    "loguru>=0.7.3",
-    "lz4>=4.4.3",
-    "requests>=2.32.3",
-]
-classifiers = [
-    "Development Status :: 5 - Production/Stable",
-    "Intended Audience :: Developers",
-    "Intended Audience :: Science/Research",
-    "Topic :: Scientific/Engineering :: Artificial Intelligence",
-    "Programming Language :: Python :: 3.12",
-    "Programming Language :: Python :: 3.13",
-    "Programming Language :: Python :: 3.14",
-]
-
-[project.optional-dependencies]
-audio = [
-    "torch>=2.11,<2.12",
-    "torchaudio>=2.11,<2.12",
-]
-torch = [
-    "torch>=2.6.0",
-    "timm>=1.0.15",
-]
-language = [
-    "tensorflow>=2.18.1",
-    "tensorflow-text>=2.18.1",
-]
-clip = [
-    "ftfy>=6.1,<7",
-    "regex>=2024.11.6",
-]
-extras = [
-    "matplotlib>=3.10.1",
-    "pillow>=11.1.0",
-    "scikit-learn>=1.6.1",
-]
-
-[build-system]
-requires = ["uv_build>=0.10.12,<0.12.0"]
-build-backend = "uv_build"
-
-[tool.uv.build-backend]
-source-include = [
-    "CHANGELOG.md",
-    "docs/licensing/sam3-provenance.md",
-    "docs/licensing/t0-source-provenance.md",
-    "docs/migration-v2.md",
-    "docs/stability.md",
-    "docs/timeseries.md",
-    "models/huggingface/NOTICE",
-    "models/huggingface/README.md",
-    "models/huggingface/convnext-conversion.json",
-]
-wheel-exclude = ["EGM.md"]
-
-[tool.ruff.lint]
-ignore = [
-    "E731",
-    "E741",
-    "F722",
-]
-
-[tool.ty.analysis]
-replace-imports-with-any = ["equinox._module._module"]
-allowed-unresolved-imports = [
-    "ftfy.**",
-    "matplotlib.**",
-    "PIL.**",
-    "regex.**",
-    "sklearn.**",
-    "tensorflow.**",
-    "tensorflow_text.**",
-    "timm.**",
-    "torch.**",
-    "torchaudio.**",
-]
-
-[tool.pytest.ini_options]
-addopts = "-m 'not reference_parity and not live_reference_parity'"
-markers = [
-    "reference_parity: parity against checked-in outputs from pinned upstream implementations",
-    "live_reference_parity: parity requiring a separately supplied upstream checkpoint",
-]
-
-[dependency-groups]
-dev = [
-    "jax2onnx>=0.17.0",
-    "mdformat==1.0.0",
-    "mdformat-footnote==0.1.3",
-    "mdformat-gfm==1.0.0",
-    "mdformat-myst==0.3.0",
-    "onnxruntime>=1.28.0",
-    "pre-commit>=4.5.1",
-    "pytest>=9.0.2",
-    "ruff>=0.15.7",
-    "ty>=0.0.24",
-]
-reference = [
-    "tabpfn>=8.1.0",
-    "tfc-t0==0.2.1",
-    "timm>=1.0.15",
-    "torch>=2.6.0",
-    "transformers>=5.13.1",
-]

equimo-2.8.0/src/equimo/__init__.py

@@ -1 +0,0 @@
-__version__ = "2.8.0"