yury-lysogorskiy commited on
Commit
d81e55b
·
verified ·
1 Parent(s): 6df56b5

v4 (uqv6) GRACE-3L release: fp32 models, checkpoints, kokkos, changelog

Browse files
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:1636ec2a87e7cbfc0cbb46f5b55af117fb1fab271765652a9633aa6492f38e3d
3
- size 156714415
 
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