File size: 7,002 Bytes
fb0d523 | 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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 | ---
tags:
- Biology
- RNA Design
- Inverse Folding
- Geometric Deep Learning
license: agpl-3.0
library_name: ofoldx
pipeline_tag: other
---
# gRNAde
Pre-trained models for RNA inverse folding from one or more backbone conformations.
## Disclaimer
This is an UNOFFICIAL implementation of the [gRNAde: Geometric Deep Learning for 3D RNA inverse design](https://arxiv.org/abs/2305.14749) by Chaitanya K. Joshi, et al.
The OFFICIAL repository of gRNAde is at [chaitjo/geometric-rna-design](https://github.com/chaitjo/geometric-rna-design).
**The team releasing gRNAde did not write this model card for this model so this model card has been written by the OFoldX team.**
## Model Details
gRNAde uses a geometric-vector perceptron (GVP) encoder and an autoregressive nucleotide decoder for RNA inverse folding.
The encoder combines one or more aligned conformations before predicting RNA identities. The v1 graph incorporates
sequence order, explicit secondary-structure base pairs, and optional P, C4', and base-nitrogen geometry; the v0.3.2 graph
uses spatial geometry without sequence or base-pair edge channels.
Please refer to the [Training Details](#training-details) section for more information on the training process.
### Variants
- **v1**: [gRNAde](https://huggingface.co/oteam/grnade) and the [ODesign checkpoint](https://huggingface.co/oteam/grnade-odesign).
- **v0.3.2, 1 state**: [all](https://huggingface.co/oteam/grnade-v0.3.2-1state-all),
[das](https://huggingface.co/oteam/grnade-v0.3.2-1state-das), [multi](https://huggingface.co/oteam/grnade-v0.3.2-1state-multi).
- **v0.3.2, 2 states**: [all](https://huggingface.co/oteam/grnade-v0.3.2-2state-all).
- **v0.3.2, 3 states**: [all](https://huggingface.co/oteam/grnade-v0.3.2-3state-all),
[das](https://huggingface.co/oteam/grnade-v0.3.2-3state-das), [multi](https://huggingface.co/oteam/grnade-v0.3.2-3state-multi).
- **v0.3.2, 5 states**: [all](https://huggingface.co/oteam/grnade-v0.3.2-5state-all),
[das](https://huggingface.co/oteam/grnade-v0.3.2-5state-das), [multi](https://huggingface.co/oteam/grnade-v0.3.2-5state-multi).
### Model Specification
| Variant | Encoder Blocks | Decoder Blocks | Node Scalar / Vector Size | Edge Scalar / Vector Size | Input Edge Channels | Num Parameters (M) |
| ------- | -------------- | -------------- | ------------------------- | ------------------------- | ------------------- | ------------------ |
| v1 | 4 | 4 | 128 / 16 | 64 / 4 | 132 | 2.157 |
| v0.3.2 | 4 | 4 | 128 / 16 | 64 / 4 | 131 | 2.157 |
### Links
- **Code**: [ofoldx.models.grnade](https://github.com/OTeam-AI4S/OFoldX/tree/main/ofoldx/models/grnade)
- **Paper**: [gRNAde: Geometric Deep Learning for 3D RNA inverse design](https://arxiv.org/abs/2305.14749);
[Generative inverse design of RNA structure and function with gRNAde](https://www.biorxiv.org/content/10.1101/2025.11.29.691298)
- **Developed by**: Chaitanya K. Joshi, Arian R. Jamasb, Ramon Viñas, Charles Harris, Simon V. Mathis, Alex Morehead,
Rishabh Anand, and Pietro Liò
- **Model type**: Multi-conformation GVP encoder with an autoregressive RNA sequence decoder
- **Original Repository**: [chaitjo/geometric-rna-design](https://github.com/chaitjo/geometric-rna-design)
## Usage
The model file depends on the [`ofoldx`](https://ofoldx.valtech.ai/) library. You can install it using pip:
```bash
pip install ofoldx
```
### Direct Use
#### RNA Sequence Design
You can use this model directly with a pipeline for RNA sequence design from a backbone and secondary structure:
```python
import ofoldx
from ofoldx.io import read_structure
structure, _ = read_structure("rna.cif")
pipeline = ofoldx.pipeline("design", model="oteam/grnade")
result = pipeline(structure, secondary_structure="((((....))))", num_candidates=4, temperature=0.2)
```
The dot-bracket string covers the selected RNA chain; use `instance_id` when the structure contains several RNA instances.
Each candidate contains its molecular target, design record, and sequence scores.
### Downstream Use
#### Multi-Conformation Design
You can condition one sequence design on multiple aligned conformations of the same RNA:
```python
import ofoldx
from ofoldx.io import read_structure
open_structure, _ = read_structure("rna_open.cif")
closed_structure, _ = read_structure("rna_closed.cif")
pipeline = ofoldx.pipeline("design", model="oteam/grnade")
result = pipeline((open_structure, closed_structure), secondary_structure="((((....))))")
```
The conformations must contain the same RNA sequence. The first is the design target; later conformations supply
structural context.
## Training Details
gRNAde is trained to recover RNA nucleotide identities from one or more backbone conformations.
### Training Data
The gRNAde authors' [RNASolo dataset](https://huggingface.co/datasets/chaitjo/gRNAde_datasets) contains RNA structures
from the Protein Data Bank at 4 Å resolution or better, with a 31 October 2023 cutoff.
### Training Procedure
gRNAde uses autoregressive RNA sequence recovery from structural context. The released label-smoothing value was `0.05`.
OFoldX provides [`GrnadeLoss`][ofoldx.models.grnade.GrnadeLoss] for training and fine-tuning.
## Citation
```bibtex
@inproceedings{joshi2025grnade,
title={g{RNA}de: Geometric Deep Learning for 3D RNA inverse design},
author={Joshi, Chaitanya K and Jamasb, Arian R and Vi{\~n}as, Ramon and
Harris, Charles and Mathis, Simon V and Morehead, Alex and
Anand, Rishabh and Li{\`o}, Pietro},
booktitle={International Conference on Learning Representations},
year={2025}
}
```
> [!NOTE]
> The artifacts distributed in this repository are part of the OFoldX project.
> If OFoldX supports your research, please cite the OFoldX project as follows:
```bibtex
@software{chen2026ofoldx,
author = {Chen, Zhiyuan},
title = {OFoldX},
year = {2026},
url = {https://github.com/OTeam-AI4S/OFoldX},
}
```
## Contact
Please use GitHub issues of [OFoldX](https://github.com/OTeam-AI4S/OFoldX/issues) for any questions or comments on the model implementation or model card.
Please contact the authors of the [gRNAde paper](https://arxiv.org/abs/2305.14749) for questions or comments on the paper/model.
## License
The model weights, implementation, configuration, and documentation distributed by OFoldX are licensed under the
[GNU Affero General Public License](https://github.com/OTeam-AI4S/OFoldX/blob/main/license.md).
For additional terms and clarifications, please refer to our [License FAQ](https://github.com/OTeam-AI4S/OFoldX/blob/main/license-faq.md).
The upstream gRNAde source is MIT-licensed.
```spdx
SPDX-License-Identifier: AGPL-3.0-or-later
```
[ofoldx.models.grnade.GrnadeLoss]: https://ofoldx.valtech.ai/models/grnade/#ofoldx.models.grnade.GrnadeLoss
|