Re-issue 2L SavedModels with compute_dense (model-v3-uq-v6 one-time reset)

CHANGELOG.md

@@ -13,6 +13,29 @@ Newest release on top.
 
 ---
 
+## v3 — uq-v6 — model tag RE-ISSUED 2026-07-06 (add `compute_dense` to 2L)
+
+- **Deliberate, one-time reset of the immutable `model-v3-uq-v6` tag** (protocol §4 tags are
+  normally never moved; done here on explicit request). Only the **SavedModel** payload
+  changed — `checkpoint-v3-uq-v6` and `kk-v3-uq-v6` are **untouched** (checkpoints and kokkos
+  are byte-identical; the re-export was made *from* those same checkpoints).
+- **What changed:** the **10 2L** models (OMAT/SMAX/SMAX-OMAT, medium+large), both fp32 and
+  `-fp64` = **20 SavedModels**, re-exported so the SavedModel now also carries the
+  **`compute_dense`** signature (fast dense-reshape neighbor engine) beside `compute`
+  (segment_sum). Enables `TPCalculator(..., mode="uniform")` / the dense engine on these models.
+- **1L unchanged (8 models):** 1L is a local, scalar-SPBF architecture with **no** equivariant
+  dense engine, so `compute_dense` does not apply — the 1L SavedModels under the tag are the
+  same v3 artifacts.
+- Re-export reused the existing uqv6 GMM artifacts (no retrain, no GMM rebuild). Weights
+  identical: re-export vs the previous SavedModel agrees to machine precision (fp64 dE=0,
+  dF~1e-15; fp32 pure summation-reorder round-off ≤5e-6). `compute` vs `compute_dense` E/F
+  agree (fp64 dE=0). fp64 dirs re-exported with `GRACE_UQ_LENIENT_LOAD=1` (old-format
+  checkpoints carry one extra non-weight bool flag; validated safe against the deployed fp64).
+- **Client URLs unchanged** — `foundation_models.py` already pins `model-v3-uq-v6`; the shifted
+  tag serves the compute_dense 2L SavedModels transparently.
+
+---
+
 ## v4 — uq-v6 — *DEPLOYED 2026-07-06*
 
 - First **3-layer (3L)** GRACE foundation models, shipped with UQ:

models/GRACE-2L-OMAT-large-base-fp64-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7cf07301a267a526a14fcf56bc06c18a7973e4a2806b4b7b66444b1de9137b60
-size 232938488
+oid sha256:4c6b48bbb7591996771735812f31663e55a9fc7dbb7210793d3f5c12cef3d924
+size 233053145

models/GRACE-2L-OMAT-large-base-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f2402a0f7ca32623d8a54e8a753530350578ce568284cd642838b0a4ee248df7
-size 121214607
+oid sha256:28514a07e186a8e246fc046404be9bdc32073e42312ab8c5005e46eef53a49c0
+size 121321900

models/GRACE-2L-OMAT-large-ft-AM-fp64-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:af30bb045d5f89aefa73c5c322b313009cb69ca4bb1f22a8e9c5ff0780fb6491
-size 237074461
+oid sha256:b23aaf81fb835caaca78200b94961f81601aff4869f1c517b41818a76adcfdef
+size 237188842

models/GRACE-2L-OMAT-large-ft-AM-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7518e67392d19a5c729a029ab55525fe462d981b973b3096230b1c42a089b8e9
-size 124449770
+oid sha256:a27926838a844bc95e6c8ad3287b4bc275f8d72b9b9bbafc83ede59a95f85894
+size 124558444

models/GRACE-2L-OMAT-large-ft-E-fp64-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:732928212b80dad4a9936536e34b1ae42e7ec3251c3acbf338c7b48af5b24150
-size 232943027
+oid sha256:869bcb88502cbbb83886720b1d42a9c8816a71e6d03d0587bcf0f19a624d2e56
+size 233053490

models/GRACE-2L-OMAT-large-ft-E-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:86e718af2fd23361b16861504c0fa0bfff865f0e3f58eb7d0922b8df1107373d
-size 121216090
+oid sha256:b943ae9abdb4b9e55a297e0a9d5010200d0e651e0d8f960dcd7acebca501a02f
+size 121323360

models/GRACE-2L-OMAT-medium-base-fp64-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:92985b0bed61458e2206afaea917eabafa459d775115cb02333a3a382e19c612
-size 197962073
+oid sha256:bcdf43b4820c073009b276c47535bafc32c70ef9a7a9e43a7117c5c3ab056a83
+size 198076148

models/GRACE-2L-OMAT-medium-base-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:472143e3383e136d6de7e022307cf95524c38b83ccaae435b16990ce15227576
-size 104608990
+oid sha256:19eb065b421532bdd3af2abf39443745bca2059a15601e289e4e888f08677682
+size 104719178

models/GRACE-2L-OMAT-medium-ft-AM-fp64-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:30023173c6aee6751d84a8d5931f0fcb97afe9e1a447137f3eecf6bd1cbbcf69
-size 202772662
+oid sha256:d9fd6fd7207335dba7308d7a0dddb9c8543dc960e4ec3c6da8366529ee0ea2d9
+size 202901456

models/GRACE-2L-OMAT-medium-ft-AM-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a5b2b5283a90ac3c0f5f3a3a9c52b8d728c71dde43f590be6413385f9f970e7b
-size 108532062
+oid sha256:edaf7ecf08ecc5b7a7a985bc96f8ff7f657fc9faff92704792a54051e457afc3
+size 108643260

models/GRACE-2L-OMAT-medium-ft-E-fp64-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ba363faaaf2e5949f17ebee44d922eedf4b0864f51ba2474fd57baaaef1a85e2
-size 197960923
+oid sha256:dcdd4374032a3e557339a5ef587f411eaf5a2ef73e7a9b59c8b9ec8357063a26
+size 198077236

models/GRACE-2L-OMAT-medium-ft-E-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0d8413cb41605a9948beb4bdbfb3d7388013ca61b319818d4a1a97db5eb2f5b8
-size 104610393
+oid sha256:662c55ecafb90a4b84bbfbc68561354725a64e185ad7d7661831b63d14843726
+size 104722041

models/GRACE-2L-SMAX-OMAT-large-fp64-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:71957e3c13ac4e01284f03770d30015a4d316fd6280d5573946214f1434ae405
-size 246653742
+oid sha256:bf6f8c3b429324462946ca27a3a68adfd89c4369e2debcd279f816b2d0212129
+size 246766954

models/GRACE-2L-SMAX-OMAT-large-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:75dbf8bb7d38cac5fac410fa5fe54536f1ef86c896f5d5f8e7139037ab7f9845
-size 128600451
+oid sha256:cce4dee8b18b85585e21a6299ee26fe26e11920e1a9835aea67c5b70e2e3f66f
+size 128718366

models/GRACE-2L-SMAX-OMAT-medium-fp64-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4099cb0221fba8573915da57eecf225681e987ae340d26f19ebc71d0ec20b794
-size 209710455
+oid sha256:3af84bb53d715738ba4c05fb76b096b701cdb0db8c19c32ee35e0c17cb6cbd23
+size 209824803

models/GRACE-2L-SMAX-OMAT-medium-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:807122f43008848c0b2b28e3c53e4b3638d878e96e1f095513841d69aae891b1
-size 111057745
+oid sha256:e83747f1abb0582e0344721c4e40196850114b9b216539de5355dfed770ee39c
+size 111167474

models/GRACE-2L-SMAX-large-fp64-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fd701e988ee2f9ad9a57e6e98fcde1049aa15963fd49bc17886a4041da4ee5cb
-size 236363547
+oid sha256:238281899648dfa30e2771860d32f9d4373473e7b695b81d81d7d7392d1fedba
+size 236475774

models/GRACE-2L-SMAX-large-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c12f0feccb2c0f3f91ab10f0c3851e3e8e09a01cf166fee65cc1f6dba1ce19ce
-size 124863025
+oid sha256:7f339f14886f23f9e161c33797c0511302921e09504f6c7b240cc31ad3328cb8
+size 124978359

models/GRACE-2L-SMAX-medium-fp64-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9a2b130d12c11688d9133a50cd38b6f660e86aa66b99bdca73c628d85e97d135
-size 199061461
+oid sha256:e21f96e4090a5049b16e4e208dedb9a3fd720cd27eee0a882180d0280cc7f4ad
+size 199177183

models/GRACE-2L-SMAX-medium-model.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3e87c0537cca7a46bdf6d3d628a90edcfade77c5e88edfd72b54318854d7ac7e
-size 106261554
+oid sha256:9edd5adbf4b89c3535f4ba257e097c2e7262789622a87e201f337bbf69a887ef
+size 106370236