GRACE-2L-OMAT, GRACE-2L-OAM: layer-norm epsilon in kokkos.npz (format 3) and checkpoint model.yaml

CHANGELOG.md

@@ -13,6 +13,23 @@ Newest release on top.
 
 ---
 
+## kk-legacy-v2, checkpoint-legacy-v2 — no UQ — *2026-11-11* (layer-norm epsilon of GRACE-2L-OMAT and GRACE-2L-OAM)
+
+- **GRACE-2L-OMAT and GRACE-2L-OAM only:** `kokkos/GRACE-2L-OMAT-kokkos.npz`, `kokkos/GRACE-2L-OAM-kokkos.npz`,
+  `checkpoints/GRACE-2L-OMAT-checkpoint.tar.gz` and `checkpoints/GRACE-2L-OAM-checkpoint.tar.gz`. Their SavedModels
+  are unchanged (tag `kk`). GRACE-1L-OMAT and GRACE-1L-OAM are unchanged and keep `kk-legacy-v1`.
+- **What changed:** the epsilon of the RMS layer norm of the energy-MLP input, 1e-12 as in the SavedModels, is now stored
+  in the checkpoint `model.yaml` (`ln_epsilon`) and in `kokkos.npz` (`energy_mlp_ln_eps`). grace-runtime, LAMMPS and a
+  TF model built from the checkpoint now match the SavedModel when every bond of an atom is near the 6 Å cutoff.
+  All weights are unchanged.
+- **`kokkos.npz` format 3:** needs grace-runtime newer than 0.1.4; older versions refuse the files and ask for an
+  update. The checkpoints need tensorpotential newer than 0.6.2.1 for the setting to take effect; older versions
+  ignore it and use 1e-16, as before.
+- **No UQ.**
+- Tags: `kk-legacy-v2`, `checkpoint-legacy-v2`. Earlier tags are untouched.
+
+---
+
 ## kk-legacy-v1 — no UQ — *2026-10-31* (kokkos.npz of the legacy GRACE-1L/2L models)
 
 - **Kokkos only, outside the `vX` release numbering:** `kokkos/GRACE-1L-OAM-kokkos.npz`,

checkpoints/GRACE-2L-OAM-checkpoint.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:04f3be4e99d5a8c9bbbe1803facd30f739e3e907c910d9687b93dadb8d8cee6f
-size 97154775
+oid sha256:9b1dd2fe5efc76de00edbbc4204ac472c65d1596b3697c024fb44fc260237587
+size 97154758

checkpoints/GRACE-2L-OMAT-checkpoint.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8d58862b1bd93a034e6ce9d4d483e8377069d94c30960d50bc1c871171d6bf97
-size 97114635
+oid sha256:00538458c5fc1bb815329dac4e75a049c9ce2332f2020c66b574d6a8b414016e
+size 97114461

kokkos/GRACE-2L-OAM-kokkos.npz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:86ea29271df85f8955c89d6f376e335781b5423957617b550b1a5ca38e69c261
-size 100940908
+oid sha256:975b26b82b4314c16298979f326a7de6f851159f003c0ab94c82db5c5b4c9961
+size 100941182

kokkos/GRACE-2L-OMAT-kokkos.npz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:37501f76d03d61293a43a3c4a756977271391ad54d58105aecb7662e0ae051e9
-size 100940908
+oid sha256:65e4f3a1b15097057ae42d4aa3f996ab13a1c3b8c994eb8c0985d21ff53a0140
+size 100941182