v4 (uqv6) GRACE-3L release: fp32 models, checkpoints, kokkos, changelog
Browse files- CHANGELOG.md +21 -0
- checkpoints/GRACE-3L-OMAT-large-checkpoint.tar.gz +3 -0
- checkpoints/GRACE-3L-OMAT-large-ft-AM-checkpoint.tar.gz +3 -0
- kokkos/GRACE-3L-OMAT-large-ft-AM-kokkos.npz +3 -0
- kokkos/GRACE-3L-OMAT-large-kokkos.npz +3 -0
- models/GRACE-3L-OMAT-large-ft-AM-model.tar.gz +2 -2
- models/GRACE-3L-OMAT-large-model.tar.gz +3 -0
CHANGELOG.md
CHANGED
|
@@ -13,6 +13,27 @@ Newest release on top.
|
|
| 13 |
|
| 14 |
---
|
| 15 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
## v3 — uq-v6 — *DEPLOYED 2026-06-26*
|
| 17 |
|
| 18 |
- All 18 foundation models rebuilt with the **uqv6** UQ feature (schema 6:
|
|
|
|
| 13 |
|
| 14 |
---
|
| 15 |
|
| 16 |
+
## v4 — uq-v6 — *DEPLOYED 2026-07-06*
|
| 17 |
+
|
| 18 |
+
- First **3-layer (3L)** GRACE foundation models, shipped with UQ:
|
| 19 |
+
**`GRACE-3L-OMAT-large`** (base, fitted on OMat24) and
|
| 20 |
+
**`GRACE-3L-OMAT-large-ft-AM`** (fine-tuned on sAlex+MPTraj).
|
| 21 |
+
- **uqv6** UQ feature (schema 6, basis-RP **D=132**, normalize + log-norm density);
|
| 22 |
+
per-atom gamma flags both compression and stretch.
|
| 23 |
+
- **fp32-only.** These models are natively fp32 (no fp64 build exists), so the bare
|
| 24 |
+
name *is* the fp32 payload — there is **no `-fp64` variant** (unlike the 18 v3
|
| 25 |
+
2L/1L models). kokkos.npz is float-promoted (fp64 storage), served separately.
|
| 26 |
+
- **LAMMPS kk = `pair_style grace/3l/kk[/fp32]`** — MIXED/FP32 precision only (there
|
| 27 |
+
is no fp64 3L kk kernel). Requires the schema-6-aware build with `grace/3l/kk`
|
| 28 |
+
compiled in (its cuBLAS-sgemm selftest threshold is sized for fp32 GEMM). KK-vs-TF
|
| 29 |
+
agreement is therefore at fp32 tolerance (E ≤5e-7 eV/atom, F ≤1.5e-6 eV/Å,
|
| 30 |
+
gamma rel ≤3e-5), not the ~1e-8 of the fp64 2L/1L kk.
|
| 31 |
+
- Validated (2026-07-06): KK-vs-TF (E/F/stress + gamma, both models) + quick E-nn
|
| 32 |
+
(gamma bounded, no compression blow-up).
|
| 33 |
+
- Tags: `model-v4-uq-v6`, `checkpoint-v4-uq-v6`, `kk-v4-uq-v6`.
|
| 34 |
+
- Supersedes the earlier upload-only, non-UQ `model-v3` ft-AM SavedModel (left in place;
|
| 35 |
+
immutable).
|
| 36 |
+
|
| 37 |
## v3 — uq-v6 — *DEPLOYED 2026-06-26*
|
| 38 |
|
| 39 |
- All 18 foundation models rebuilt with the **uqv6** UQ feature (schema 6:
|
checkpoints/GRACE-3L-OMAT-large-checkpoint.tar.gz
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:aae0f9f67ce3de6d3a6d2d9b36057bb07494f9fd431e35c9b7d7c9ab02def7aa
|
| 3 |
+
size 214619982
|
checkpoints/GRACE-3L-OMAT-large-ft-AM-checkpoint.tar.gz
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b4b6fd1c91f69efe08c8caafabbf024b4dd4bdf082974296c1a0f2f28c7f9a56
|
| 3 |
+
size 223789872
|
kokkos/GRACE-3L-OMAT-large-ft-AM-kokkos.npz
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:10d390ebb8d4ee36bd9ea3df268327dbb92de598fabcec40d0e6a93085422916
|
| 3 |
+
size 441377086
|
kokkos/GRACE-3L-OMAT-large-kokkos.npz
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9f62dd08e5adf1fd0dc9ae56383694bc7073f4690f58a6d2ac10dd5bb6a67d6a
|
| 3 |
+
size 397625042
|
models/GRACE-3L-OMAT-large-ft-AM-model.tar.gz
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ede0fc88af9ed54fee2358d342f6f5a38a1b4b5b2e3ec78ae408c10315ce0f40
|
| 3 |
+
size 191368412
|
models/GRACE-3L-OMAT-large-model.tar.gz
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4fcf257af042fa1723b620ddef22c3569486e438d641303da239dafcac1a8a33
|
| 3 |
+
size 186739417
|