README.md 660 Bytes
Newer Older
Gustaf Ahdritz's avatar
Gustaf Ahdritz committed
1
2
3
4
5
6
7
# OpenFold

A faithful PyTorch reproduction of DeepMind's 
[AlphaFold 2](https://github.com/deepmind/alphafold).

## Installation

Gustaf Ahdritz's avatar
Gustaf Ahdritz committed
8
1. Having installed Python 3.9, install dependencies.
Gustaf Ahdritz's avatar
Gustaf Ahdritz committed
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

```bash
pip3 install -r requirements.txt
```

2. Install third-party dependencies.

```bash
scripts/install_third_party_dependencies.py
```

## Features

OpenFold fully reproduces AlphaFold's inference and data processing pipelines, 
but is written entirely in PyTorch rather than JAX/TensorFlow. It is capable of 
importing AlphaFold's original pretrained weights.

Future versions will support multi-GPU training with 
[DeepSpeed](https://github.com/microsoft/DeepSpeed).