Hugging Face's logo

AethronPhantom
/
Nexa_Mat2

PyTorch
Safetensors
materials-science
crystal-generation
diffusion
encoder
multimodal
gradio
Model card Files
xet
 Community
Allanatrix commited on
Commit
fd43acc
·
verified ·
1 Parent(s): e0030d0

chore: publish safetensors NexaMat stack artifacts

Browse files
Files changed (1) hide show
  1. encoder/v1/nexa_mat_V1_final.safetensors.json +411 -0
encoder/v1/nexa_mat_V1_final.safetensors.json ADDED
@@ -0,0 +1,411 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "source_pt": "/tmp/nexa_mat_weights_hf/encoder/v1/nexa_mat_V1_final.pt",
3
+ "safetensors": "/tmp/nexa_mat_weights_hf/encoder/v1/nexa_mat_V1_final.safetensors",
4
+ "bytes": 11264456,
5
+ "sha256": "a39e1db3c86622cc36bc2610a47af0f6336578fc3b644a7068ed62190853e9a0",
6
+ "tensor_count": 100,
7
+ "step": 200000,
8
+ "metadata": {
9
+ "created_at_utc": "2026-05-06T07:50:37Z",
10
+ "backend": "materials",
11
+ "training_plan": "aetheron_materials_encoder_pretrain_v1",
12
+ "training_phase": "encoder_pretrain",
13
+ "model_family": "directional_potential",
14
+ "model_config": {
15
+ "hidden_dim": 192,
16
+ "egnn_layers": 4,
17
+ "max_neighbors": 8,
18
+ "energy_mode": "atomwise_sum",
19
+ "use_absolute_coords": false,
20
+ "radial_basis": 64,
21
+ "cutoff_angstrom": 6.0,
22
+ "pair_energy_scale": 1.5,
23
+ "angular_energy_scale": 1.0,
24
+ "atom_energy_scale": 1.0,
25
+ "element_reference_scale": 0.0,
26
+ "message_aggregation": "learned"
27
+ },
28
+ "dataset_root": "/workspace/nexa_mat_v1",
29
+ "resumed_from": "/workspace/nexa_compute/artifacts/nexa_mat/runs/nexa_mat_encoder_blackwell_full_v1_20260505T204743Z/final_checkpoint.pt",
30
+ "start_step": 164159,
31
+ "dataset_manifest": {
32
+ "dataset_manifest_path": null,
33
+ "dataset_manifest_sha256": null,
34
+ "phase_receipt_path": "/workspace/nexa_mat_v1/phase_receipt.json",
35
+ "phase_receipt_sha256": "sha256:a98d4ffb76473013420c65a9a79d07b2940ccfa31f52ebea90c17942b41ac8dc",
36
+ "phase_ready": true
37
+ },
38
+ "distributed": {
39
+ "rank": 0,
40
+ "world_size": 1
41
+ },
42
+ "train_split": "train",
43
+ "val_split": "val",
44
+ "train_shards": [
45
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00012-part00000.parquet",
46
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00019-part00000.parquet",
47
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00008-part00000.parquet",
48
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00000-part00000.parquet",
49
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00013-part00000.parquet",
50
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00015-part00000.parquet",
51
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00009-part00000.parquet",
52
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00010-part00000.parquet",
53
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00012-part00000.parquet",
54
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00014-part00000.parquet",
55
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00009-part00000.parquet",
56
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00011-part00000.parquet",
57
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00005-part00000.parquet",
58
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00000-part00000.parquet",
59
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00001-part00000.parquet",
60
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00015-part00000.parquet",
61
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00016-part00000.parquet",
62
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00000-part00000.parquet",
63
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00000-part00000.parquet",
64
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00002-part00000.parquet",
65
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00008-part00000.parquet",
66
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00000-part00000.parquet",
67
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00013-part00000.parquet",
68
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00014-part00000.parquet",
69
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00005-part00000.parquet",
70
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00007-part00000.parquet",
71
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00005-part00000.parquet",
72
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00004-part00000.parquet",
73
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00019-part00000.parquet",
74
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00014-part00000.parquet",
75
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00006-part00000.parquet",
76
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00003-part00000.parquet",
77
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00001-part00000.parquet",
78
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00009-part00000.parquet",
79
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00004-part00000.parquet",
80
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00011-part00000.parquet",
81
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00011-part00000.parquet",
82
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00001-part00000.parquet",
83
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00017-part00000.parquet",
84
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00002-part00000.parquet",
85
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00008-part00000.parquet",
86
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00014-part00000.parquet",
87
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00011-part00000.parquet",
88
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00014-part00000.parquet",
89
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00002-part00000.parquet",
90
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00009-part00000.parquet",
91
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00007-part00000.parquet",
92
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00002-part00000.parquet",
93
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00011-part00000.parquet",
94
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00012-part00000.parquet",
95
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00013-part00000.parquet",
96
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00012-part00000.parquet",
97
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00011-part00000.parquet",
98
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00010-part00000.parquet",
99
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00008-part00000.parquet",
100
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00014-part00000.parquet",
101
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00013-part00000.parquet",
102
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00019-part00000.parquet",
103
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00016-part00000.parquet",
104
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00016-part00000.parquet",
105
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00006-part00000.parquet",
106
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00005-part00000.parquet",
107
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00011-part00000.parquet",
108
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00008-part00000.parquet",
109
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00016-part00000.parquet",
110
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00012-part00000.parquet",
111
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00001-part00000.parquet",
112
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00015-part00000.parquet",
113
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00003-part00000.parquet",
114
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00012-part00000.parquet",
115
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00009-part00000.parquet",
116
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00005-part00000.parquet",
117
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00010-part00000.parquet",
118
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00016-part00000.parquet",
119
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00015-part00000.parquet",
120
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00001-part00000.parquet",
121
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00005-part00000.parquet",
122
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00000-part00000.parquet",
123
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00018-part00000.parquet",
124
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00009-part00000.parquet",
125
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00019-part00000.parquet",
126
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00012-part00000.parquet",
127
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00001-part00000.parquet",
128
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00004-part00000.parquet",
129
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00015-part00000.parquet",
130
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00009-part00000.parquet",
131
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00012-part00000.parquet",
132
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00013-part00000.parquet",
133
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00015-part00000.parquet",
134
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00006-part00000.parquet",
135
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00014-part00000.parquet",
136
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00010-part00000.parquet",
137
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00018-part00000.parquet",
138
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00014-part00000.parquet",
139
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00012-part00000.parquet",
140
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00005-part00000.parquet",
141
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00013-part00000.parquet",
142
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00005-part00000.parquet",
143
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00003-part00000.parquet",
144
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00003-part00000.parquet",
145
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00012-part00000.parquet",
146
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00006-part00000.parquet",
147
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00014-part00000.parquet",
148
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00014-part00000.parquet",
149
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00011-part00000.parquet",
150
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00014-part00000.parquet",
151
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00017-part00000.parquet",
152
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00003-part00000.parquet",
153
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00018-part00000.parquet",
154
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00011-part00000.parquet",
155
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00008-part00000.parquet",
156
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00017-part00000.parquet",
157
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00012-part00000.parquet",
158
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00003-part00000.parquet",
159
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00000-part00000.parquet",
160
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00013-part00000.parquet",
161
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00006-part00000.parquet",
162
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00004-part00000.parquet",
163
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00004-part00000.parquet",
164
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00002-part00000.parquet",
165
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00004-part00000.parquet",
166
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00010-part00000.parquet",
167
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00010-part00000.parquet",
168
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00010-part00000.parquet",
169
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00006-part00000.parquet",
170
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00007-part00000.parquet",
171
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00019-part00000.parquet",
172
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00015-part00000.parquet",
173
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00006-part00000.parquet",
174
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00019-part00000.parquet",
175
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00000-part00000.parquet",
176
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00002-part00000.parquet",
177
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00004-part00000.parquet",
178
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00001-part00000.parquet",
179
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00018-part00000.parquet",
180
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00009-part00000.parquet",
181
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00002-part00000.parquet",
182
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00008-part00000.parquet",
183
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00016-part00000.parquet",
184
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00018-part00000.parquet",
185
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00001-part00000.parquet",
186
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00004-part00000.parquet",
187
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00007-part00000.parquet",
188
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00006-part00000.parquet",
189
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00011-part00000.parquet",
190
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00017-part00000.parquet",
191
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00013-part00000.parquet",
192
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00006-part00000.parquet",
193
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00009-part00000.parquet",
194
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00007-part00000.parquet",
195
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00003-part00000.parquet",
196
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00018-part00000.parquet",
197
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00009-part00000.parquet",
198
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00017-part00000.parquet",
199
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00016-part00000.parquet",
200
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00002-part00000.parquet",
201
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00013-part00000.parquet",
202
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00018-part00000.parquet",
203
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00004-part00000.parquet",
204
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00000-part00000.parquet",
205
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00017-part00000.parquet",
206
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00013-part00000.parquet",
207
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00007-part00000.parquet",
208
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00015-part00000.parquet",
209
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00016-part00000.parquet",
210
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00018-part00000.parquet",
211
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00016-part00000.parquet",
212
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00001-part00000.parquet",
213
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00008-part00000.parquet",
214
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00019-part00000.parquet",
215
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00010-part00000.parquet",
216
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00001-part00000.parquet",
217
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00019-part00000.parquet",
218
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00010-part00000.parquet",
219
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00007-part00000.parquet",
220
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00015-part00000.parquet",
221
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00007-part00000.parquet",
222
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00007-part00000.parquet",
223
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00002-part00000.parquet",
224
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00005-part00000.parquet",
225
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00010-part00000.parquet",
226
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00002-part00000.parquet",
227
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00008-part00000.parquet",
228
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00009-part00000.parquet",
229
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00000-part00000.parquet",
230
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00013-part00000.parquet",
231
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00019-part00000.parquet",
232
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00017-part00000.parquet",
233
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00008-part00000.parquet",
234
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00015-part00000.parquet",
235
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00017-part00000.parquet",
236
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00008-part00000.parquet",
237
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00018-part00000.parquet",
238
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00018-part00000.parquet",
239
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00007-part00000.parquet",
240
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00016-part00000.parquet",
241
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00019-part00000.parquet",
242
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00010-part00000.parquet",
243
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00005-part00000.parquet",
244
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00000-part00000.parquet",
245
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00016-part00000.parquet",
246
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00015-part00000.parquet",
247
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00017-part00000.parquet",
248
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00007-part00000.parquet",
249
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00011-part00000.parquet",
250
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00001-part00000.parquet",
251
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00017-part00000.parquet",
252
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00006-part00000.parquet",
253
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00018-part00000.parquet",
254
+ "/workspace/nexa_mat_v1/train/rattled-1000/omat24-00019-part00000.parquet",
255
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-npt/omat24-00003-part00000.parquet",
256
+ "/workspace/nexa_mat_v1/train/rattled-300/omat24-00004-part00000.parquet",
257
+ "/workspace/nexa_mat_v1/train/rattled-500-subsampled/omat24-00004-part00000.parquet",
258
+ "/workspace/nexa_mat_v1/train/rattled-300-subsampled/omat24-00003-part00000.parquet",
259
+ "/workspace/nexa_mat_v1/train/rattled-relax/omat24-00003-part00000.parquet",
260
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-npt/omat24-00003-part00000.parquet",
261
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-3000-nvt/omat24-00002-part00000.parquet",
262
+ "/workspace/nexa_mat_v1/train/rattled-500/omat24-00005-part00000.parquet",
263
+ "/workspace/nexa_mat_v1/train/aimd-from-PBE-1000-nvt/omat24-00017-part00000.parquet",
264
+ "/workspace/nexa_mat_v1/train/rattled-1000-subsampled/omat24-00006-part00000.parquet"
265
+ ],
266
+ "val_shards": [
267
+ "/workspace/nexa_mat_v1/val/aimd-from-PBE-1000-npt/omat24-00020-part00000.parquet",
268
+ "/workspace/nexa_mat_v1/val/aimd-from-PBE-1000-npt/omat24-00021-part00000.parquet",
269
+ "/workspace/nexa_mat_v1/val/aimd-from-PBE-1000-nvt/omat24-00020-part00000.parquet",
270
+ "/workspace/nexa_mat_v1/val/aimd-from-PBE-1000-nvt/omat24-00021-part00000.parquet",
271
+ "/workspace/nexa_mat_v1/val/aimd-from-PBE-3000-npt/omat24-00020-part00000.parquet",
272
+ "/workspace/nexa_mat_v1/val/aimd-from-PBE-3000-npt/omat24-00021-part00000.parquet",
273
+ "/workspace/nexa_mat_v1/val/aimd-from-PBE-3000-nvt/omat24-00020-part00000.parquet",
274
+ "/workspace/nexa_mat_v1/val/aimd-from-PBE-3000-nvt/omat24-00021-part00000.parquet",
275
+ "/workspace/nexa_mat_v1/val/rattled-1000/omat24-00020-part00000.parquet",
276
+ "/workspace/nexa_mat_v1/val/rattled-1000/omat24-00021-part00000.parquet",
277
+ "/workspace/nexa_mat_v1/val/rattled-1000-subsampled/omat24-00020-part00000.parquet",
278
+ "/workspace/nexa_mat_v1/val/rattled-1000-subsampled/omat24-00021-part00000.parquet",
279
+ "/workspace/nexa_mat_v1/val/rattled-300/omat24-00020-part00000.parquet",
280
+ "/workspace/nexa_mat_v1/val/rattled-300/omat24-00021-part00000.parquet",
281
+ "/workspace/nexa_mat_v1/val/rattled-300-subsampled/omat24-00020-part00000.parquet",
282
+ "/workspace/nexa_mat_v1/val/rattled-300-subsampled/omat24-00021-part00000.parquet",
283
+ "/workspace/nexa_mat_v1/val/rattled-500/omat24-00020-part00000.parquet",
284
+ "/workspace/nexa_mat_v1/val/rattled-500/omat24-00021-part00000.parquet",
285
+ "/workspace/nexa_mat_v1/val/rattled-500-subsampled/omat24-00020-part00000.parquet",
286
+ "/workspace/nexa_mat_v1/val/rattled-500-subsampled/omat24-00021-part00000.parquet",
287
+ "/workspace/nexa_mat_v1/val/rattled-relax/omat24-00020-part00000.parquet",
288
+ "/workspace/nexa_mat_v1/val/rattled-relax/omat24-00021-part00000.parquet"
289
+ ],
290
+ "objective": {
291
+ "encoder_pretrain": {
292
+ "physics_anchor": {
293
+ "energy_per_atom_l1_weight": 1.0,
294
+ "force_mse_weight": 12.0,
295
+ "stress_l1_weight": 0.05
296
+ },
297
+ "contrastive_weight": 0.0,
298
+ "denoise_weight": 0.0,
299
+ "broad_property_weight": 0.0,
300
+ "contrastive_temperature": 0.1,
301
+ "force_loss": "smooth_l1",
302
+ "force_norm_weight": 2.0,
303
+ "force_cosine_weight": 2.5,
304
+ "force_norm_ratio_weight": 5.0,
305
+ "force_norm_ratio_min_true_mean": 0.05,
306
+ "force_norm_ratio_clamp": 4.0,
307
+ "force_source_norm_ratio_weight": 0.0,
308
+ "force_source_norm_ratio_min_sites": 8,
309
+ "force_tail_norm_ratio_weight": 1.0,
310
+ "force_tail_norm_ratio_quantile": 0.9,
311
+ "direct_force_aux_weight": 4.0,
312
+ "direct_gradient_consistency_weight": 0.1,
313
+ "force_norm_floor": 0.01,
314
+ "force_tail_weight": 0.15,
315
+ "force_tail_power": 1.0,
316
+ "force_tail_weight_cap": 2.0,
317
+ "force_record_balance": true,
318
+ "force_source_policy": {
319
+ "enabled": true,
320
+ "allow_regex": "^omat24(:|$)",
321
+ "deny_regex": "",
322
+ "scope": "force_labels_only",
323
+ "note": "When enabled, force masks are filtered by source/regime before loss and force metrics. Energy, stress, and source telemetry remain unfiltered."
324
+ },
325
+ "energy_shift": 0.0,
326
+ "energy_scale": 1.0,
327
+ "force_scale": 1.0,
328
+ "force_huber_delta": 0.5,
329
+ "force_target": "energy_gradient",
330
+ "optimizer": {
331
+ "name": "adamw",
332
+ "lr": 0.0001,
333
+ "weight_decay": 0.01,
334
+ "max_grad_norm": 1.0,
335
+ "grad_accum_steps": 1
336
+ },
337
+ "lr_schedule": {
338
+ "name": "cosine",
339
+ "warmup_steps": 1000,
340
+ "min_lr_ratio": 0.1
341
+ },
342
+ "schedule": {
343
+ "name": "static",
344
+ "energy_warmup_steps": 0,
345
+ "force_norm_warmup_steps": 0,
346
+ "force_cosine_warmup_steps": 0,
347
+ "initial_energy_weight": -1.0,
348
+ "initial_force_norm_weight": -1.0,
349
+ "initial_force_cosine_weight": -1.0
350
+ },
351
+ "note": "Physics-supervised energy/force/stress loss remains dominant; auxiliary losses regularize the manifold."
352
+ }
353
+ },
354
+ "label_contract": {
355
+ "min_batch_physics_label_fraction": 0.95,
356
+ "min_batch_energy_label_fraction": 0.75,
357
+ "min_batch_force_label_fraction": 0.95,
358
+ "policy": "Optional fail-fast gate. Missing labels remain masked unless a configured minimum is violated."
359
+ },
360
+ "batching": {
361
+ "train_batch_size_cap": 512,
362
+ "eval_batch_size_cap": 128,
363
+ "train_atom_budget": 49152,
364
+ "train_pair_budget": 0,
365
+ "train_padded_pair_budget": 0,
366
+ "train_sort_buffer": 8192,
367
+ "precompute_neighbors": true,
368
+ "train_shard_order": "source_round_robin",
369
+ "source_interleave_rows": 8,
370
+ "require_physics_labels": true,
371
+ "require_force_labels": true,
372
+ "row_source_allow_regex": "^omat24(:|$)",
373
+ "row_source_deny_regex": "",
374
+ "eval_atom_budget": 8192,
375
+ "eval_pair_budget": 0,
376
+ "eval_padded_pair_budget": 0,
377
+ "eval_sort_buffer": 2048,
378
+ "max_atoms_per_record": 128,
379
+ "drop_oversized_records": true,
380
+ "drop_oversized_eval_records": true,
381
+ "policy": "Optional memory-aware batching. Row batching remains active when budgets are zero."
382
+ },
383
+ "observability": {
384
+ "source_stratified_eval": true,
385
+ "source_stratified_train": true,
386
+ "source_metric_min_records": 8,
387
+ "source_metric_max_sources": 24,
388
+ "perf_snapshot_every": 100,
389
+ "train_diagnostics_every": 100,
390
+ "eval_rows_per_shard": 32,
391
+ "profiled_stages": [
392
+ "dataloader",
393
+ "device_transfer",
394
+ "forward",
395
+ "force_gradient_autograd",
396
+ "loss",
397
+ "backward",
398
+ "optimizer",
399
+ "metrics"
400
+ ]
401
+ },
402
+ "run_data_receipt": {
403
+ "path": "/workspace/nexa_compute/artifacts/nexa_mat/runs/nexa_mat_encoder_blackwell_full_v1_continue_to_200k_20260506T075034Z/run_data_receipt.json",
404
+ "sha256": "sha256:092019c2ab55e561c0dfd5e3a20530b757c3e5652cfcbb0aa6aa925011d7110d",
405
+ "selected_train_shards": 220,
406
+ "selected_val_shards": 22,
407
+ "train_bytes": 140754508987,
408
+ "val_bytes": 14178627058
409
+ }
410
+ }
411
+ }