VizFold

Composable infrastructure for protein structure prediction, tracing, and interpretability on HPC systems.

Supporting ESMFold, Boltz-2, intermediate representation extraction, and reproducible deployment workflows.

About VizFold

VizFold is a research initiative aimed at developing advanced visualization and interpretability techniques for AlphaFold and similar protein folding models. Our work bridges computational science and biology to enhance understanding of protein folding mechanisms.

Core Features

Attention Extraction

Capture and analyze model attention patterns across protein sequences.

Intermediate Representations

Expose embeddings, pair representations, and internal model states for analysis.

Arc Visualizations

Generate arc-diagram views for long-range residue relationships.

HPC Deployment

Support reproducible execution across cluster and high-performance environments.

Framework Architecture

VizFold separates model execution, representation extraction, visualization, and HPC deployment into composable workflow components.

VizFold framework architecture

Supported Backends

ESMFold

Fast protein structure prediction with support for representation tracing.

Boltz-2

Modern structure prediction backend integrated into reproducible workflows.

HPC Workflows

Deployment patterns for running prediction and analysis pipelines on clusters.

Presented at PEARC 2026

A Composable and Modular Framework for Protein Structure Prediction on HPC

VizFold introduces a modular infrastructure for running protein structure prediction workflows, extracting intermediate representations, and supporting interpretability analysis on HPC systems.

Project Team

Dr. Giri Krishnan - PI, Georgia Tech

Dr. Polo Chau - Co-PI, Georgia Tech

Dr. Tyler Hayes Georgia Tech

Suresh Marru Georgia Tech

Contribute to VizFold

We welcome researchers, developers, and contributors! Check out our GitHub repository for ways to get involved.