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

Pytorch implementation of MalConv

Real-time object detection on Android using the YOLO network with TensorFlow

An open source library for face detection in images. The face detection speed can reach 1000FPS.

SEC'21: Sparse Bitmap Compression for Memory-Efficient Training onthe Edge

【Arxiv】Exploring Separable Attention for Multi-Contrast MR Image Super-Resolution

A Text Attention Network for Spatial Deformation Robust Scene Text Image Super-resolution (CVPR2022)

Making self-supervised learning work on molecules by using their 3D geometry to pre-train GNNs. Implemented in DGL and Pytorch Geometric.

A clean and scalable template to kickstart your deep learning project 🚀 ⚡ 🔥

SwinTrack: A Simple and Strong Baseline for Transformer Tracking

OpenABC-D: A Large-Scale Dataset For Machine Learning Guided Integrated Circuit Synthesis

[IJCAI-2021] A benchmark of data-free knowledge distillation from paper "Contrastive Model Inversion for Data-Free Knowledge Distillation"

Implementation for paper "STAR: A Structure-aware Lightweight Transformer for Real-time Image Enhancement" (ICCV 2021).

Ranking Models in Unlabeled New Environments （iccv21）

On-device speech-to-intent engine powered by deep learning

BED: A Real-Time Object Detection System for Edge Devices

Keras + Hyperopt: A very simple wrapper for convenient hyperparameter optimization

Code of our paper "Contrastive Object-level Pre-training with Spatial Noise Curriculum Learning"

iPOKE: Poking a Still Image for Controlled Stochastic Video Synthesis

Implementation of Convolutional LSTM in PyTorch.

Python implementation of Bayesian optimization over permutation spaces.