Skip to content

RosettaCommons/modelforge

Repository files navigation

Open-Source Neural Networks for Biomolecular Tasks

ModelForge is a repository of open-source models for common biomolecular tasks, including structure prediction, fixed-backbone sequence design ("inverse folding"), and de novo protein design.

All models within ModelForge share a common training harness and integrate with AtomWorks – our generalized computational framework for biomolecular modeling.

For more information, please see our preprint, Accelerating Biomolecular Modeling with AtomWorks and RF3.

⚠️ Notice: We fixed an inference bug on 8/29 that arose during codebase migration and impacted predictions from JSON and from mmCIF/PDB; the issue is now resolved but for the purposes of model benchmarking predictions should be re-run.

⚠️ Notice: We are currently finalizing some cleanup work within our repositories. Please expect the APIs (e.g., function and class names, inputs and outputs) to stabilize within the next two weeks. Thank you for your patience!

⚠️ Notice: Training code coming very soon, with documentation on how to fine-tune on new datasets!

Supported Networks

RosettaFold3 (RF3)

RF3 is a structure prediction neural network that narrows the gap between closed-source AF-3 and open-source alternatives.

Protein-DNA complex prediction

Complete inference instructions for RF3 are provided here.

Installation & Usage

Follow these steps to set up ModelForge and run a test prediction.


1. Install the repository using uv

git clone https://github.com/RosettaCommons/modelforge.git \
  && cd modelforge \
  && uv python install 3.12 \
  && uv venv --python 3.12 \
  && source .venv/bin/activate \
  && uv pip install -e .

2. Download model weights for RF3

wget http://files.ipd.uw.edu/pub/rf3/rf3_latest.pt

3. Run a test prediction

rf3 fold tests/data/5vht_from_json.json

Details on the exact formatting of the json files are available here.

About

Central repository for biomolecular foundation models with shared trainers and pipeline components

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •  

Languages