Open Catalyst 2020 Nudged Elastic Band (OC20NEB)

Open Catalyst 2020 Nudged Elastic Band (OC20NEB)#

Overview#

This is a validation dataset which was used to assess model performance in CatTSunami: Accelerating Transition State Energy Calculations with Pre-trained Graph Neural Networks. It is comprised of 932 NEB relaxation trajectories. There are three different types of reactions represented: desorptions, dissociations, and transfers. NEB calculations allow us to find transition states. The rate of reaction is determined by the transition state energy, so access to transition states is very important for catalysis research. For more information, check out the paper.

File Structure and Contents#

The tar file contains 3 subdirectories: dissociations, desorptions, and transfers. As the names imply, these directories contain the converged DFT trajectories for each of the reaction classes. Within these directories, the trajectories are named to identify the contents of the file. Here is an example and the anatomy of the name:

desorption_id_83_2409_9_111-4_neb1.0.traj

  1. desorption indicates the reaction type (dissociation and transfer are the other possibilities)

  2. id identifies that the material belongs to the validation in domain split (ood - out of domain is th e other possibility)

  3. 83 is the task id. This does not provide relavent information

  4. 2409 is the bulk index of the bulk used in the ocdata bulk pickle file

  5. 9 is the reaction index. for each reaction type there is a reaction pickle file in the repository. In this case it is the 9th entry to that pickle file

  6. 111-4 the first 3 numbers are the miller indices (i.e. the (1,1,1) surface), and the last number cooresponds to the shift value. In this case the 4th shift enumerated was the one used.

  7. neb1.0 the number here indicates the k value used. For the full dataset, 1.0 was used so this does not distiguish any of the trajectories from one another.

The content of these trajectory files is the repeating frame sets. Despite the initial and final frames not being optimized during the NEB, the initial and final frames are saved for every iteration in the trajectory. For the dataset, 10 frames were used - 8 which were optimized over the neb. So the length of the trajectory is the number of iterations (N) * 10. If you wanted to look at the frame set prior to optimization and the optimized frame set, you could get them like this:

from __future__ import annotations

!wget https://dl.fbaipublicfiles.com/opencatalystproject/data/large_files/desorption_id_83_2409_9_111-4_neb1.0.traj

from ase.io import read

traj = read("desorption_id_83_2409_9_111-4_neb1.0.traj", ":")
unrelaxed_frames = traj[0:10]
relaxed_frames = traj[-10:]
--2025-06-18 14:34:08--  https://dl.fbaipublicfiles.com/opencatalystproject/data/large_files/desorption_id_83_2409_9_111-4_neb1.0.traj
Resolving dl.fbaipublicfiles.com (dl.fbaipublicfiles.com)... 
18.238.192.99, 18.238.192.60, 18.238.192.122, ...
Connecting to dl.fbaipublicfiles.com (dl.fbaipublicfiles.com)|18.238.192.99|:443... connected.
HTTP request sent, awaiting response... 
200 OK
Length: 10074935 (9.6M) [binary/octet-stream]
Saving to: ‘desorption_id_83_2409_9_111-4_neb1.0.traj’


          desorptio   0%[                    ]       0  --.-KB/s               
         desorption  83%[===============>    ]   8.00M  29.1MB/s               
desorption_id_83_24 100%[===================>]   9.61M  33.9MB/s    in 0.3s    

2025-06-18 14:34:08 (33.9 MB/s) - ‘desorption_id_83_2409_9_111-4_neb1.0.traj’ saved [10074935/10074935]

Download#

Splits

Size of compressed version (in bytes)

Size of uncompressed version (in bytes)

MD5 checksum (download link)

ASE Trajectories

1.5G

6.3G

52af34a93758c82fae951e52af445089

Use#

One more note: We have not prepared an lmdb for this dataset. This is because it is NEB calculations are not supported directly in ocp. You must use the ase native OCP class along with ase infrastructure to run NEB calculations. Here is an example of a use:

from ase.io import read
from ase.mep import DyNEB
from ase.optimize import BFGS
from fairchem.core import FAIRChemCalculator, pretrained_mlip

traj = read("desorption_id_83_2409_9_111-4_neb1.0.traj", ":")
images = traj[0:10]
predictor = pretrained_mlip.get_predict_unit("uma-s-1")

neb = DyNEB(images, k=1)
for image in images:
    image.calc = FAIRChemCalculator(predictor, task_name="oc20")

optimizer = BFGS(
    neb,
    trajectory="neb.traj",
)

conv = optimizer.run(fmax=0.45, steps=200)
if conv:
    neb.climb = True
    conv = optimizer.run(fmax=0.05, steps=300)
/home/runner/work/_tool/Python/3.12.11/x64/lib/python3.12/site-packages/torchtnt/utils/version.py:12: UserWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html. The pkg_resources package is slated for removal as early as 2025-11-30. Refrain from using this package or pin to Setuptools<81.
  import pkg_resources
WARNING:root:device was not explicitly set, using device='cuda'.
      Step     Time          Energy          fmax
BFGS:    0 14:34:24     -305.751997        5.179585
BFGS:    1 14:34:25     -305.678596       11.366769
BFGS:    2 14:34:26     -305.904274        1.882657
BFGS:    3 14:34:27     -305.919916        2.602309
BFGS:    4 14:34:28     -305.997625        2.277943
BFGS:    5 14:34:29     -305.991819        6.859359
BFGS:    6 14:34:31     -306.241671        9.557948
BFGS:    7 14:34:32     -306.214908        3.379573
BFGS:    8 14:34:33     -306.278919        4.739883
BFGS:    9 14:34:34     -306.303228        0.742368
BFGS:   10 14:34:35     -306.317829        0.645264
BFGS:   11 14:34:36     -306.347292        1.625757
BFGS:   12 14:34:37     -306.403383        2.013868
BFGS:   13 14:34:39     -306.437923        0.595247
BFGS:   14 14:34:40     -306.408136        0.705192
BFGS:   15 14:34:41     -306.445282        1.372755
BFGS:   16 14:34:42     -306.486189        0.854076
BFGS:   17 14:34:43     -306.484566        0.428320
BFGS:   18 14:34:44     -306.281544        9.125462
BFGS:   19 14:34:45     -306.460533        1.417865
BFGS:   20 14:34:47     -306.463280        1.051390
BFGS:   21 14:34:48     -306.351564        2.334730
BFGS:   22 14:34:49     -306.270340        3.786557
BFGS:   23 14:34:50     -306.294540        1.427301
BFGS:   24 14:34:51     -306.188530        3.115916
BFGS:   25 14:34:52     -305.962940        5.878541
BFGS:   26 14:34:53     -305.710041        6.851119
BFGS:   27 14:34:54     -305.448500        6.567751
BFGS:   28 14:34:56     -305.190399        5.392535
BFGS:   29 14:34:57     -304.935068        3.842170
BFGS:   30 14:34:58     -304.671394        2.734126
BFGS:   31 14:34:59     -304.411434        2.780117
BFGS:   32 14:35:00     -304.189175        2.618870
BFGS:   33 14:35:01     -304.019625        3.095736
BFGS:   34 14:35:03     -303.909305        4.322395
BFGS:   35 14:35:04     -303.800432        5.058592
BFGS:   36 14:35:05     -303.683979        5.492490
BFGS:   37 14:35:06     -303.510622        5.698408
BFGS:   38 14:35:07     -303.315550        5.807897
BFGS:   39 14:35:08     -303.042068        5.850423
BFGS:   40 14:35:09     -302.717363        5.877101
BFGS:   41 14:35:10     -302.362955        5.930020
BFGS:   42 14:35:12     -301.965973        5.940001
BFGS:   43 14:35:13     -301.552250        5.985605
BFGS:   44 14:35:14     -301.110157        6.146882
BFGS:   45 14:35:15     -300.776688        6.385990
BFGS:   46 14:35:16     -300.972778        6.034767
BFGS:   47 14:35:17     -300.861561        5.046716
BFGS:   48 14:35:18     -300.803415        4.100813
BFGS:   49 14:35:20     -300.670116        4.691816
BFGS:   50 14:35:21     -300.663177        4.289832
BFGS:   51 14:35:22     -300.663807        4.562983
BFGS:   52 14:35:23     -300.595198        3.833212
BFGS:   53 14:35:24     -300.511551        3.854820
BFGS:   54 14:35:25     -300.375988        4.079554
BFGS:   55 14:35:26     -300.233013        4.118357
BFGS:   56 14:35:27     -300.080198        4.044491
BFGS:   57 14:35:29     -299.928619        3.896747
BFGS:   58 14:35:30     -299.784265        3.694881
BFGS:   59 14:35:31     -299.646690        3.467095
BFGS:   60 14:35:32     -299.504589        3.249350
BFGS:   61 14:35:33     -299.348434        3.140039
BFGS:   62 14:35:34     -299.190132        3.343379
BFGS:   63 14:35:35     -299.070000        3.591612
BFGS:   64 14:35:37     -299.027872        3.672315
BFGS:   65 14:35:38     -299.069954        4.430005
BFGS:   66 14:35:39     -299.181742        5.001501
BFGS:   67 14:35:40     -299.305044        5.382055
BFGS:   68 14:35:41     -299.389391        5.668052
BFGS:   69 14:35:42     -299.407738        5.691513
BFGS:   70 14:35:43     -299.375423        5.593985
BFGS:   71 14:35:44     -299.321178        5.417364
BFGS:   72 14:35:46     -299.262230        5.205491
BFGS:   73 14:35:47     -299.207491        4.995058
BFGS:   74 14:35:48     -299.146803        4.804477
BFGS:   75 14:35:49     -299.072637        4.645102
BFGS:   76 14:35:50     -298.985914        4.508088
BFGS:   77 14:35:51     -298.886721        4.371045
BFGS:   78 14:35:52     -298.771337        4.236062
BFGS:   79 14:35:54     -298.655952        4.083562
BFGS:   80 14:35:55     -298.564400        3.930572
BFGS:   81 14:35:56     -298.497480        3.766889
BFGS:   82 14:35:57     -298.451507        3.621720
BFGS:   83 14:35:58     -298.419142        3.475254
BFGS:   84 14:35:59     -298.392364        3.371453
BFGS:   85 14:36:00     -298.364973        3.524727
BFGS:   86 14:36:02     -298.331762        4.029002
BFGS:   87 14:36:03     -298.286466        4.447284
BFGS:   88 14:36:04     -298.227676        4.867840
BFGS:   89 14:36:05     -298.150305        5.189667
BFGS:   90 14:36:06     -298.055921        5.506477
BFGS:   91 14:36:07     -297.940906        5.764251
BFGS:   92 14:36:08     -297.804714        5.989054
BFGS:   93 14:36:10     -297.646170        6.170612
BFGS:   94 14:36:11     -297.463913        6.314513
BFGS:   95 14:36:12     -297.256404        6.419111
BFGS:   96 14:36:13     -297.022742        6.486026
BFGS:   97 14:36:14     -296.762841        6.517731
BFGS:   98 14:36:15     -296.476454        6.518214
BFGS:   99 14:36:16     -296.163617        6.484715
BFGS:  100 14:36:18     -295.825180        6.424899
BFGS:  101 14:36:19     -295.462803        6.340613
BFGS:  102 14:36:20     -295.078546        6.231403
BFGS:  103 14:36:21     -294.674727        6.101545
BFGS:  104 14:36:22     -294.254717        5.950826
BFGS:  105 14:36:23     -293.823242        5.779688
BFGS:  106 14:36:24     -293.386684        5.592644
BFGS:  107 14:36:26     -292.952050        5.393141
BFGS:  108 14:36:27     -292.524431        5.187044
BFGS:  109 14:36:28     -292.109192        4.976302
BFGS:  110 14:36:29     -291.713185        4.785156
BFGS:  111 14:36:30     -291.339586        4.843581
BFGS:  112 14:36:31     -290.985568        4.911133
BFGS:  113 14:36:32     -290.642259        4.977214
BFGS:  114 14:36:34     -290.300152        5.032003
BFGS:  115 14:36:35     -289.953740        5.070847
BFGS:  116 14:36:36     -289.600652        5.093106
BFGS:  117 14:36:37     -289.239069        5.099436
BFGS:  118 14:36:38     -288.866993        5.090187
BFGS:  119 14:36:39     -288.483854        5.064533
BFGS:  120 14:36:41     -288.090595        5.020142
BFGS:  121 14:36:42     -287.688314        4.953238
BFGS:  122 14:36:43     -287.278039        4.960165
BFGS:  123 14:36:44     -286.861767        5.124053
BFGS:  124 14:36:45     -286.442058        5.272077
BFGS:  125 14:36:46     -286.021759        5.403789
BFGS:  126 14:36:47     -285.603523        5.520214
BFGS:  127 14:36:49     -285.188380        5.631916
BFGS:  128 14:36:50     -284.777136        5.789363
BFGS:  129 14:36:51     -284.392787        5.887674
BFGS:  130 14:36:52     -284.010064        5.934777
BFGS:  131 14:36:53     -283.632913        6.023956
BFGS:  132 14:36:54     -283.262353        6.110120
BFGS:  133 14:36:55     -282.898248        6.194447
BFGS:  134 14:36:57     -282.540691        6.276842
BFGS:  135 14:36:58     -282.101758        6.375357
BFGS:  136 14:36:59     -281.695464        6.458852
BFGS:  137 14:37:00     -281.307849        6.531172
BFGS:  138 14:37:01     -280.941184        6.681297
BFGS:  139 14:37:02     -280.594590        6.931446
BFGS:  140 14:37:04     -280.267860        7.158994
BFGS:  141 14:37:05     -279.957745        7.348140
BFGS:  142 14:37:06     -279.661869        7.491121
BFGS:  143 14:37:07     -279.377719        7.573011
BFGS:  144 14:37:08     -279.105585        7.563769
BFGS:  145 14:37:09     -278.848191        7.415536
BFGS:  146 14:37:10     -278.613659        7.063231
BFGS:  147 14:37:12     -278.405995        7.172911
BFGS:  148 14:37:13     -278.231431        7.342077
BFGS:  149 14:37:14     -278.093239        7.527584
BFGS:  150 14:37:15     -277.986935        7.758623
BFGS:  151 14:37:16     -277.917101        8.064655
BFGS:  152 14:37:17     -277.884477        8.413619
BFGS:  153 14:37:18     -277.883565        8.851371
BFGS:  154 14:37:20     -277.904317        9.304932
BFGS:  155 14:37:21     -277.955784        9.832757
BFGS:  156 14:37:22     -278.023983       10.122661
BFGS:  157 14:37:23     -278.118318       10.219844
BFGS:  158 14:37:24     -278.250482        9.988359
BFGS:  159 14:37:25     -278.429600        9.697521
BFGS:  160 14:37:27     -278.665860        9.383339
BFGS:  161 14:37:28     -278.956453        9.095147
BFGS:  162 14:37:29     -279.285358        8.855175
BFGS:  163 14:37:30     -279.633726        9.526032
BFGS:  164 14:37:31     -279.979620       10.075938
BFGS:  165 14:37:33     -280.299614       10.420508
BFGS:  166 14:37:34     -280.588268       10.603022
BFGS:  167 14:37:35     -280.830742       10.658484
BFGS:  168 14:37:36     -281.013079       10.646706
BFGS:  169 14:37:37     -281.124248       10.572913
BFGS:  170 14:37:38     -281.160971       10.461680
BFGS:  171 14:37:39     -281.111747       10.321001
BFGS:  172 14:37:41     -280.992872       10.152760
BFGS:  173 14:37:42     -280.817468        9.963685
BFGS:  174 14:37:43     -280.598428        9.766789
BFGS:  175 14:37:44     -280.353240        9.564391
BFGS:  176 14:37:45     -280.103510        9.350678
BFGS:  177 14:37:46     -279.962004        9.049541
BFGS:  178 14:37:48     -279.893787        9.221679
BFGS:  179 14:37:49     -280.084705        9.348490
BFGS:  180 14:37:50     -280.376070        9.521127
BFGS:  181 14:37:51     -280.471366        9.575539
BFGS:  182 14:37:52     -280.464377        9.620071
BFGS:  183 14:37:53     -280.399441        9.659474
BFGS:  184 14:37:54     -280.184905        9.706169
BFGS:  185 14:37:56     -279.798408        9.723506
BFGS:  186 14:37:57     -279.255155        9.709318
BFGS:  187 14:37:58     -278.626530        9.675148
BFGS:  188 14:37:59     -277.949592        9.630718
BFGS:  189 14:38:00     -277.194444        9.565775
BFGS:  190 14:38:01     -276.377630        9.476157
BFGS:  191 14:38:03     -275.642245        9.356320
BFGS:  192 14:38:04     -274.996846        9.223189
BFGS:  193 14:38:05     -274.445835        9.088888
BFGS:  194 14:38:06     -273.841928        8.957783
BFGS:  195 14:38:07     -273.303872        8.791905
BFGS:  196 14:38:08     -272.858058        8.657247
BFGS:  197 14:38:10     -272.409642        8.470602
BFGS:  198 14:38:11     -271.982956        8.397922
BFGS:  199 14:38:12     -271.581454        8.257087
BFGS:  200 14:38:13     -271.189741        8.010430
BFGS:  201 14:38:14     -271.048777        7.618147
BFGS:  202 14:38:16     -271.117348        7.356795
BFGS:  203 14:38:17     -271.246765        7.509110
BFGS:  204 14:38:18     -271.368793        7.607163
BFGS:  205 14:38:19     -271.498010        7.709455
BFGS:  206 14:38:20     -271.699728        7.838984
BFGS:  207 14:38:21     -271.964604        7.968263
BFGS:  208 14:38:23     -272.263061        8.085011
BFGS:  209 14:38:24     -272.586657        8.084403
BFGS:  210 14:38:25     -272.915183        8.054060
BFGS:  211 14:38:26     -273.281582        7.908693
BFGS:  212 14:38:27     -273.701825        8.030814
BFGS:  213 14:38:28     -274.205115        8.359205
BFGS:  214 14:38:30     -274.861142        9.181370
BFGS:  215 14:38:31     -275.764403       10.280328
BFGS:  216 14:38:32     -276.582500       10.408885
BFGS:  217 14:38:33     -277.488142       10.822386
BFGS:  218 14:38:34     -278.131040       10.062499
BFGS:  219 14:38:35     -278.483250        9.476072
BFGS:  220 14:38:37     -278.939478        9.323689
BFGS:  221 14:38:38     -279.228961        9.706339
BFGS:  222 14:38:39     -279.425662       10.373426
BFGS:  223 14:38:40     -279.661333       10.806636
BFGS:  224 14:38:41     -280.283510       11.082577
BFGS:  225 14:38:42     -280.860143       11.034770
BFGS:  226 14:38:44     -280.691462       11.566847
BFGS:  227 14:38:45     -279.917855       12.052370
BFGS:  228 14:38:46     -279.134258       12.296504
BFGS:  229 14:38:47     -278.436746       12.422442
BFGS:  230 14:38:48     -278.034049       12.371940
BFGS:  231 14:38:49     -277.979044       12.259945
BFGS:  232 14:38:51     -278.250655       12.124573
BFGS:  233 14:38:52     -278.388100       12.057904
BFGS:  234 14:38:53     -278.166474       11.978041
BFGS:  235 14:38:54     -277.697212       11.966547
BFGS:  236 14:38:55     -277.212114       12.002120
BFGS:  237 14:38:57     -276.786794       12.131158
BFGS:  238 14:38:58     -276.306053       12.161356
BFGS:  239 14:38:59     -275.786016       12.149778
BFGS:  240 14:39:00     -275.266892       12.146047
BFGS:  241 14:39:01     -274.733916       12.215017
BFGS:  242 14:39:02     -274.230104       12.242084
BFGS:  243 14:39:04     -273.803747       12.172743
BFGS:  244 14:39:05     -273.571749       11.666065
BFGS:  245 14:39:06     -273.313960       11.282742
BFGS:  246 14:39:07     -273.709343       10.546539
BFGS:  247 14:39:08     -274.543423        9.917477
BFGS:  248 14:39:09     -274.626395        9.571817
BFGS:  249 14:39:11     -274.157844        9.451679
BFGS:  250 14:39:12     -273.586667        9.620856
BFGS:  251 14:39:13     -272.912583        9.656736
BFGS:  252 14:39:14     -272.389512        9.684677
BFGS:  253 14:39:15     -272.020875        9.719091
BFGS:  254 14:39:16     -271.666489       11.087652
BFGS:  255 14:39:18     -271.487815       11.124374
BFGS:  256 14:39:19     -271.463331       11.171039
BFGS:  257 14:39:20     -271.487554       11.154497
BFGS:  258 14:39:21     -271.874413       11.134522
BFGS:  259 14:39:22     -272.566259       10.811517
BFGS:  260 14:39:23     -273.734818       10.781848
BFGS:  261 14:39:25     -274.220876       10.477770
BFGS:  262 14:39:26     -274.458358       10.505292
BFGS:  263 14:39:27     -274.567471       13.986198
BFGS:  264 14:39:28     -273.817635       18.132024
BFGS:  265 14:39:29     -272.106557       22.743114
BFGS:  266 14:39:30     -271.935886       25.229473
BFGS:  267 14:39:32     -272.579899       26.254481
BFGS:  268 14:39:33     -273.334465       26.377122
BFGS:  269 14:39:34     -273.951318       25.352365
BFGS:  270 14:39:35     -274.533764       24.028669
BFGS:  271 14:39:36     -275.067366       20.615566
BFGS:  272 14:39:37     -275.628189       18.165747
BFGS:  273 14:39:39     -276.172210       16.562812
BFGS:  274 14:39:40     -276.511970       15.368924
BFGS:  275 14:39:41     -276.578775       14.019692
BFGS:  276 14:39:42     -276.676292       13.642705
BFGS:  277 14:39:43     -276.480681       13.555614
BFGS:  278 14:39:45     -276.126171       12.092261
BFGS:  279 14:39:46     -275.920549       12.279351
BFGS:  280 14:39:47     -276.337877       12.296310
BFGS:  281 14:39:48     -277.610026       10.635877
BFGS:  282 14:39:49     -277.180584       10.329642
BFGS:  283 14:39:51     -276.745185       10.100150
BFGS:  284 14:39:52     -276.472645        9.896514
BFGS:  285 14:39:53     -276.240210        9.677580
BFGS:  286 14:39:54     -276.051915        9.433171
BFGS:  287 14:39:55     -275.944962        9.542423
BFGS:  288 14:39:56     -275.892876        9.544060
BFGS:  289 14:39:58     -275.903897        9.541152
BFGS:  290 14:39:59     -275.963469        9.754072
BFGS:  291 14:40:00     -276.039730        9.980710
BFGS:  292 14:40:01     -276.072159       10.127743
BFGS:  293 14:40:02     -276.112040       10.252981
BFGS:  294 14:40:03     -276.178656       10.376428
BFGS:  295 14:40:05     -276.276811       10.457043
BFGS:  296 14:40:06     -276.046029       11.497249
BFGS:  297 14:40:07     -275.571247       11.324128
BFGS:  298 14:40:08     -275.184069       12.363022
BFGS:  299 14:40:09     -274.831624       12.608093
BFGS:  300 14:40:10     -274.687228       12.395818
BFGS:  301 14:40:12     -274.662172       11.782207
BFGS:  302 14:40:13     -274.690143       11.055881
BFGS:  303 14:40:14     -274.744181       10.778456
BFGS:  304 14:40:15     -274.856745       11.082242
BFGS:  305 14:40:16     -275.225542       11.283419
BFGS:  306 14:40:17     -276.622144       11.417250
BFGS:  307 14:40:19     -276.510848       11.337659
BFGS:  308 14:40:20     -276.094790       10.989208
BFGS:  309 14:40:21     -275.538930       10.323553
BFGS:  310 14:40:22     -275.059483        9.618803
BFGS:  311 14:40:23     -274.611047        9.961645
BFGS:  312 14:40:24     -274.215778       10.398064
BFGS:  313 14:40:26     -273.904516       11.539599
BFGS:  314 14:40:27     -273.700079       12.271411
BFGS:  315 14:40:28     -273.520143       12.583075
BFGS:  316 14:40:29     -273.364092       12.707693
BFGS:  317 14:40:30     -273.214249       12.633095