yury-lysogorskiy commited on
Commit
46ead61
·
verified ·
1 Parent(s): 454f1f3

CHANGELOG: v6 entry wording (initial/preliminary v0)

Browse files
Files changed (1) hide show
  1. CHANGELOG.md +8 -23
CHANGELOG.md CHANGED
@@ -15,32 +15,17 @@ Newest release on top.
15
 
16
  ## v6 — no UQ — *2026-09-29* (initial r²SCAN fine-tunes, v0)
17
 
18
- - **Initial fine-tune (v0) of the 2L-OMAT models on the MatPES-r²SCAN and MP-ALOE-r²SCAN
19
- datasets**, trained by **Quentin Bizot**. Two new models, each as fp32 (default) + `-fp64`:
 
20
  - **`GRACE-2L-OMAT-large-ft-mpaloe-r2scan-v0`** — `GRACE-2L-OMAT-large-base` fine-tuned on
21
- MP-ALOE (r²SCAN), 4 epochs. Test MAE 53 meV/atom (energy), 0.096 eV/Å (forces). The energy
22
- error was still falling steeply when training stopped; a longer fine-tune will follow.
23
- - **`GRACE-2L-OMAT-ft-matpes-r2scan-v0`** — the small legacy `GRACE-2L-OMAT` fine-tuned on
24
- MatPES (r²SCAN), 297 epochs (stopped at minimum learning rate). Test MAE 21 meV/atom, 0.100 eV/Å.
25
- - **r²SCAN total energies.** Both fits used `reference_energy: 0`, so they predict VASP r²SCAN
26
- total energies. Do not mix their energies with the PBE-based models.
27
  - **No UQ.** The checkpoint archives hold only `model.yaml` + `checkpoint.*` (no
28
  `gmm_artifacts.npz`), and the tags carry no `-uq-vN` suffix.
29
- - **Kokkos:** `kokkos/GRACE-2L-OMAT-large-ft-mpaloe-r2scan-v0-kokkos.npz` only (shared by both
30
- precisions). The small legacy 2L has no kokkos export.
31
- - SavedModels were re-exported from the training checkpoints (`checkpoint.best_test_loss`, optimizer
32
- state stripped) with the current code: `compute`, `compute_dense`, `compute_energy`, plus the 2L
33
- parallel split for the large model. The small legacy 2L has a single communicated key, so like
34
- its base `GRACE-2L-OMAT` it has no parallel split (`parallel: False` in the client).
35
- - Gates (2026-09-29): re-export vs the trainer's `final_model` agrees to ≤2e-10 (E/atom, F,
36
- stress, fp64); fp32 vs fp64 ≤5e-7 eV/atom, ≤1.1e-6 eV/Å; LAMMPS `grace/2l/kk` vs `pair_style
37
- grace` (fp64, Fe–Ni–O, 64 atoms) dE 2e-10 eV/atom, dF 4e-10 eV/Å.
38
- - **Known limitation: short-range holes.** Under strong isotropic compression the energy turns
39
- over and drops below the equilibrium value, as it does in the base models: e.g. for the MP-ALOE
40
- model at nearest-neighbour distances below ~1.1 Å (C, diamond), ~1.85 Å (W, bcc), ~1.7 Å
41
- (Zn, hcp). The MatPES model shows a new one in diamond C below ~1.25 Å that its base does not
42
- have. These distances are far outside equilibrium, but high-temperature or high-pressure MD
43
- can reach them.
44
  - Tags: `model-v6`, `checkpoint-v6`, `kk-v6`. Earlier tags are untouched.
45
 
46
  ---
 
15
 
16
  ## v6 — no UQ — *2026-09-29* (initial r²SCAN fine-tunes, v0)
17
 
18
+ - **Initial (preliminary, v0) fine-tune of the 2L-OMAT models on the MatPES-r²SCAN and
19
+ MP-ALOE-r²SCAN datasets**, trained by **Quentin Bizot**. Two new models, each as fp32
20
+ (default) + `-fp64`:
21
  - **`GRACE-2L-OMAT-large-ft-mpaloe-r2scan-v0`** — `GRACE-2L-OMAT-large-base` fine-tuned on
22
+ MP-ALOE.
23
+ - **`GRACE-2L-OMAT-ft-matpes-r2scan-v0`** — `GRACE-2L-OMAT` fine-tuned on MatPES.
24
+ - r²SCAN level of theory.
 
 
 
25
  - **No UQ.** The checkpoint archives hold only `model.yaml` + `checkpoint.*` (no
26
  `gmm_artifacts.npz`), and the tags carry no `-uq-vN` suffix.
27
+ - **Kokkos:** `kokkos/GRACE-2L-OMAT-large-ft-mpaloe-r2scan-v0-kokkos.npz` (shared by both
28
+ precisions of the large model).
 
 
 
 
 
 
 
 
 
 
 
 
 
29
  - Tags: `model-v6`, `checkpoint-v6`, `kk-v6`. Earlier tags are untouched.
30
 
31
  ---