Running AlphaFold2 (from ColabFold) in Azure Machine Learning

Colby T. Ford, Ph.D.

Companion repository for Medium Post: How to predict many protein structures with AlphaFold2 at-scale in Azure Machine Learning.

Resources:

Jupyter Notebook: AzureML_AlphaFold2_HyperDrive.ipynb
Sample Sequence FASTA: sequences.fasta
Dockerfile: Dockerfile

Docker Image Creation

The custom Docker image for running ColabFold in Azure ML HyperDrive is available on Docker Hub at https://hub.docker.com/r/cford38/alphafold2_aml. If you'd like to build the image for yourself, follow the instructions below.

First, build the custom Azure ML Docker image (with the required dependencies for the AlphaFold2 system).

docker build -t alphafold2_aml --build-arg IMAGE_VERSION=$(git rev-parse --short HEAD) .

# docker run --name alphafold2_aml --rm -p 8787:8787 alphafold2_aml
# docker exec -it alphafold2_aml /bin/bash

To push the image to DockerHub, run the following:

docker image tag alphafold2_aml <USERNAME>/alphafold2_aml:latest
docker push <USERNAME>/alphafold2_aml:latest

# docker image tag alphafold2_aml cford38/alphafold2_aml:latest
# docker push cford38/alphafold2_aml:latest

To push the image to an Azure Container Registry, run the following:

az login
az account set --subscription <SUBSCRIPTION_ID>

az acr login --name <CONTAINER_REGISTRY_NAME>

docker tag mmae_aml <CONTAINER_REGISTRY_NAME>.azurecr.io/alphafold2_aml:$(git rev-parse --short HEAD)
docker push <CONTAINER_REGISTRY_NAME>.azurecr.io/alphafold2_aml:$(git rev-parse --short HEAD)

Running AlphaFold2 (from ColabFold) in Azure Machine Learning

Related tags

Overview

Running AlphaFold2 (from ColabFold) in Azure Machine Learning

Colby T. Ford, Ph.D.

Resources:

Docker Image Creation

Owner

Colby T. Ford

Simple and ready-to-use tutorials for TensorFlow

Distilling Motion Planner Augmented Policies into Visual Control Policies for Robot Manipulation (CoRL 2021)

Code for STFT Transformer used in BirdCLEF 2021 competition.

SegNet-like Autoencoders in TensorFlow

PyTorch implementation of 1712.06087 "Zero-Shot" Super-Resolution using Deep Internal Learning

OntoProtein: Protein Pretraining With Ontology Embedding

Official code for Score-Based Generative Modeling through Stochastic Differential Equations

AdaMML: Adaptive Multi-Modal Learning for Efficient Video Recognition

A framework for using LSTMs to detect anomalies in multivariate time series data. Includes spacecraft anomaly data and experiments from the Mars Science Laboratory and SMAP missions.

implementation for paper "ShelfNet for fast semantic segmentation"

A script written in Python that returns a consensus string and profile matrix of a given DNA string(s) in FASTA format.

Exploring the Dual-task Correlation for Pose Guided Person Image Generation

GradAttack is a Python library for easy evaluation of privacy risks in public gradients in Federated Learning

Offcial repository for the IEEE ICRA 2021 paper Auto-Tuned Sim-to-Real Transfer.

LSTMs (Long Short Term Memory) RNN for prediction of price trends

[arXiv] What-If Motion Prediction for Autonomous Driving ❓🚗💨

[NeurIPS-2021] Slow Learning and Fast Inference: Efficient Graph Similarity Computation via Knowledge Distillation

On the Adversarial Robustness of Visual Transformer

🐥A PyTorch implementation of OpenAI's finetuned transformer language model with a script to import the weights pre-trained by OpenAI

A lane detection integrated Real-time Instance Segmentation based on YOLACT (You Only Look At CoefficienTs)