File size: 1,129 Bytes
07a254a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
Orb v3 (conservative, infinite-neighbor, OMat-trained) — JAX re-export by CuspAI
==================================================================================

This artifact is a derivative work of orb_v3_conservative_inf_omat.

Original work
-------------
Copyright 2024 Orbital Materials, Inc.
Licensed under the Apache License, Version 2.0. See LICENSE for the full
text. Upstream: https://github.com/orbital-materials/orb-models

Modifications (Apache License 2.0, §4(b))
-----------------------------------------
CuspAI converted the upstream PyTorch reference implementation of
orb_v3_conservative_inf_omat to a JAX computation graph using tojax
(https://github.com/cusp-ai-oss/tojax). The model weights and architecture
are unchanged, bit-for-bit, from the upstream release. Numerical agreement
with the PyTorch reference was verified within rtol=1e-4, atol=1e-4 during
export. The file format changed from the upstream .pt to a .zip containing
a jax.export graph plus msgpack-encoded parameters.

Attribution
-----------
Please cite Rhodes et al. (2025), arXiv:2504.06231, when using this
checkpoint in research.