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

ICON: Implicit Clothed humans Obtained from Normals (CVPR 2022)

Stacked Generative Adversarial Networks

Qlib is an AI-oriented quantitative investment platform

Circuit Training: An open-source framework for generating chip floor plans with distributed deep reinforcement learning

(Preprint) Official PyTorch implementation of "How Do Vision Transformers Work?"

Pytorch implementation of “Recursive Non-Autoregressive Graph-to-Graph Transformer for Dependency Parsing with Iterative Refinement”

PyTorch Implementation of Spatially Consistent Representation Learning(SCRL)

Python3 Implementation of (Subspace Constrained) Mean Shift Algorithm in Euclidean and Directional Product Spaces

PyTorch implementation of MSBG hearing loss model and MBSTOI intelligibility metric

Entity-Based Knowledge Conflicts in Question Answering.

DirectVoxGO reconstructs a scene representation from a set of calibrated images capturing the scene.

Fbone (Flask bone) is a Flask (Python microframework) starter/template/bootstrap/boilerplate application.

Dimension Reduced Turbulent Flow Data From Deep Vector Quantizers

Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks

[ICRA 2022] CaTGrasp: Learning Category-Level Task-Relevant Grasping in Clutter from Simulation

Real-time LIDAR-based Urban Road and Sidewalk detection for Autonomous Vehicles 🚗

The first public PyTorch implementation of Attentive Recurrent Comparators

A Python package for performing pore network modeling of porous media

Vector Quantization, in Pytorch

Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch