model card: 1L-SMAX kokkos + add -mx variants section

README.md

@@ -54,8 +54,8 @@ They offer the best balance of structural robustness (from SMAX) and high-precis
 
 | Model Name | Full Name | Size | κ_SRME | UQ | Kokkos | Description |
 | :---- | :---- | :---- | :---- | :---- | :---- | :---- |
-| GRACE-1L-SMAX-L | GRACE-1L-SMAX-large | large | 0.696 | — | — | Single-layer local (SMAX) |
-| **GRACE-1L-SMAX-OMAT-L** | **GRACE-1L-SMAX-OMAT-large** | **large** | 0.338 | — | — | **Single-layer local (SMAX + OMat24)** |
+| GRACE-1L-SMAX-L | GRACE-1L-SMAX-large | large | 0.696 | — | ✓ | Single-layer local (SMAX) |
+| **GRACE-1L-SMAX-OMAT-L** | **GRACE-1L-SMAX-OMAT-large** | **large** | 0.338 | — | ✓ | **Single-layer local (SMAX + OMat24)** |
 
 #### Two-layer, semilocal models
 
@@ -116,6 +116,17 @@ These models are first pre-trained on **OMat24** and then fine-tuned on a combin
 | GRACE-2L-OAM-M | GRACE-2L-OMAT-medium-ft-AM | medium | 0.881 | 0.200 | — | ✓ | Two-layer semi-local |
 | GRACE-2L-OAM-L | GRACE-2L-OMAT-large-ft-AM | large | 0.889 | **0.168** | — | ✓ | Two-layer semi-local |
 
+## Mixed-precision (-mx) variants
+
+Mixed-precision builds of the 2L-large models: float32 weights, which the Kokkos export
+promotes to float64. Energies and forces agree with the TF SavedModel to ~10⁻⁵ (vs ~10⁻⁷ for
+the full-precision models) — the expected effect of fp32 accumulation in the SavedModel.
+
+| Full Name | Size | UQ | Kokkos | Description |
+|:---|:---|:---|:---|:---|
+| GRACE-2L-OMAT-large-mx | large | — | ✓ | Two-layer semi-local, mixed precision (OMat24) |
+| GRACE-2L-OMAT-large-mx-ft-AM | large | — | ✓ | Two-layer semi-local, mixed precision (OMat24, fine-tuned on sAlex + MPtraj) |
+
 ---
 
 ## Repository layout