| 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. |
|
|
