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

Very simple NCHW and NHWC conversion tool for ONNX. Change to the specified input order for each and every input OP. Also, change the channel order of RGB and BGR. Simple Channel Converter for ONNX.

Unsupervised Learning of Multi-Frame Optical Flow with Occlusions

CLASP - Contrastive Language-Aminoacid Sequence Pretraining

A package related to building quasi-fibration symmetries

Source code for NAACL 2021 paper "TR-BERT: Dynamic Token Reduction for Accelerating BERT Inference"

Python library to receive live stream events like comments and gifts in realtime from TikTok LIVE.

Serverless proxy for Spark cluster

Iran Open Source Hackathon

Zsseg.baseline - Zero-Shot Semantic Segmentation

Barbershop: GAN-based Image Compositing using Segmentation Masks (SIGGRAPH Asia 2021)

Malware Analysis Neural Network project.

Code for the paper "Regularizing Variational Autoencoder with Diversity and Uncertainty Awareness"

Notes, programming assignments and quizzes from all courses within the Coursera Deep Learning specialization offered by deeplearning.ai

TensorFlowOnSpark brings TensorFlow programs to Apache Spark clusters.

The audio-video synchronization of MKV Container Format is exploited to achieve data hiding

A user-friendly research and development tool built to standardize RL competency assessment for custom agents and environments.

Object Tracking and Detection Using OpenCV

Official implementation of "A Shared Representation for Photorealistic Driving Simulators" in PyTorch.

Intrusion Detection System using ensemble learning (machine learning)

Apache Spark - A unified analytics engine for large-scale data processing