pytorch implementation of trDesign

Last update: Dec 29, 2022

Related tags

Overview

trdesign-pytorch

This repository is a PyTorch implementation of the trDesign paper based on the official TensorFlow implementation. The initial port of the trRosetta network was done by @lucidrains.

Figure 1 of De novo protein design by deep network hallucination (p. 12, Anishchenko et al., CC-BY-ND)

Requirements

Requires python 3.6+

pip install matplotlib numpy torch

Usage (protein design):

Edit src/config.py to set the experiment configuration.
Run python design.py
All results will be saved under results/

Design Configuration Options

Sequence length (int)
AA_weight (float): how strongly we want the amino acid type composition to be 'natural'
RM_AA (str): disable specific amino acid types
n_models (int): how many trRosetta model ensembles we want to use during the MCMC loop
sequence constraint (str): fix a subset of the sequence residues to specific amino acids
target_motif (path): optimize a sequence with a target motif provided as an .npz file
MCMC options

Usage (protein structure prediction):

python predict.py example.a3m
# or
python predict.py example.fasta

To get a .pdb from the resulting .npz you need to request the trRosetta package from the original authors.

Then you can run:

python trRosetta.py example.npz example.fasta output.pdb -w /tmp

References

@article {Yang1496,
  author = {Yang, Jianyi and Anishchenko, Ivan and Park, Hahnbeom and Peng, Zhenling and Ovchinnikov, Sergey and Baker, David},
  title = {Improved protein structure prediction using predicted interresidue orientations},
  year = {2020},
  doi = {10.1073/pnas.1914677117},
  URL = {https://www.pnas.org/content/117/3/1496},
  eprint = {https://www.pnas.org/content/117/3/1496.full.pdf},
  journal = {Proceedings of the National Academy of Sciences}
}

@article {Anishchenko2020.07.22.211482,
  author = {Anishchenko, Ivan and Chidyausiku, Tamuka M. and Ovchinnikov, Sergey and Pellock, Samuel J. and Baker, David},
  title = {De novo protein design by deep network hallucination},
  year = {2020},
  doi = {10.1101/2020.07.22.211482},
  URL = {https://www.biorxiv.org/content/early/2020/07/23/2020.07.22.211482},
  eprint = {https://www.biorxiv.org/content/early/2020/07/23/2020.07.22.211482.full.pdf},
  journal = {bioRxiv}
}

@article {Tischer2020.11.29.402743,
  author = {Tischer, Doug and Lisanza, Sidney and Wang, Jue and Dong, Runze and Anishchenko, Ivan and Milles, Lukas F. and Ovchinnikov, Sergey and Baker, David},
  title = {Design of proteins presenting discontinuous functional sites using deep learning},
  year = {2020},
  doi = {10.1101/2020.11.29.402743},
  URL = {https://www.biorxiv.org/content/early/2020/11/29/2020.11.29.402743},
  eprint = {https://www.biorxiv.org/content/early/2020/11/29/2020.11.29.402743.full.pdf},
  journal = {bioRxiv}
}

pytorch implementation of trDesign

Related tags

Overview

trdesign-pytorch

Requirements

Usage (protein design):

Design Configuration Options

Usage (protein structure prediction):

References

Owner

Learn Ventures Inc.

A repository for the updated version of CoinRun used to collect MUGEN, a multimodal video-audio-text dataset.

Unofficial PyTorch implementation of MobileViT based on paper "MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformer".

Image-Adaptive YOLO for Object Detection in Adverse Weather Conditions

Implementation of the final project of the course DDA6309 Probabilistic Graphical Model

OCTIS: Comparing Topic Models is Simple! A python package to optimize and evaluate topic models (accepted at EACL2021 demo track)

[NeurIPS'21] "AugMax: Adversarial Composition of Random Augmentations for Robust Training" by Haotao Wang, Chaowei Xiao, Jean Kossaifi, Zhiding Yu, Animashree Anandkumar, and Zhangyang Wang.

DA2Lite is an automated model compression toolkit for PyTorch.

[CVPR 2021] Few-shot 3D Point Cloud Semantic Segmentation

TensorFlow Metal Backend on Apple Silicon Experiments (just for fun)

DFFNet: An IoT-perceptive Dual Feature Fusion Network for General Real-time Semantic Segmentation

Optimizing Value-at-Risk and Conditional Value-at-Risk of Black Box Functions with Lacing Values (LV)

Implementation of Learning Gradient Fields for Molecular Conformation Generation (ICML 2021).

Open-Ended Commonsense Reasoning (NAACL 2021)

Classify music genre from a 10 second sound stream using a Neural Network.

Pytorch implementation of the Variational Recurrent Neural Network (VRNN).

Efficient 3D human pose estimation in video using 2D keypoint trajectories

Intelligent Video Analytics toolkit based on different inference backends.

The repo for the paper "I3CL: Intra- and Inter-Instance Collaborative Learning for Arbitrary-shaped Scene Text Detection".

Code for the SIGGRAPH 2022 paper "DeltaConv: Anisotropic Operators for Geometric Deep Learning on Point Clouds."

This is the official implementation of 3D-CVF: Generating Joint Camera and LiDAR Features Using Cross-View Spatial Feature Fusion for 3D Object Detection, built on SECOND.