Alex Morehead
amorehead
Hopper Postdoctoral Fellow @BerkeleyLab. Prev: CS PhD @BioinfoMachineLearning Lab @ UMissouri; Research intern @Profluent-AI & @AbSciBio. Deep Learning & AI4Sci
Languages
Top Repositories
Awesome papers related to generative molecular modeling and design.
Flash Attention Triton kernel with support for second-order derivatives
Implementation of AlphaFold 3 in PyTorch Lightning + Hydra
Awesome papers related to generative flow matching and its applications in bioinformatics.
An implementation of the Equivariant Graph Neural Network (EGNN) layer type for DGL-PyTorch.
Paper list for equivariant neural network
Repositories
113Personal website (based on al-folio)
PyTorch Lightning + Hydra. A very user-friendly template for rapid and reproducible ML experimentation with best practices. ⚡🔥⚡
Flash Attention Triton kernel with support for second-order derivatives
Awesome papers related to generative molecular modeling and design.
Implementation of AlphaFold 3 in PyTorch Lightning + Hydra
Awesome papers related to generative flow matching and its applications in bioinformatics.
Benchmark for generative models for materials
A Fast, Simplified Model for Molecular Generation with Improved Physical Quality
A Python web scraper written to help collect probate real estate data in an automated manner.
FAIR Chemistry's library of machine learning methods for chemistry
Platonic Transformers: A Solid Choice For Equivariance
A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes practical examples for both text and image modalities.
DockQ is a single continuous quality measure for Protein, Nucleic Acids and Small Molecule Docking Models
A lightweight library for PyTorch training tools and utilities
A collection of dot configuration files for personal use.
NeMo: a toolkit for conversational AI
Official implementation of All Atom Diffusion Transformers (ICML 2025)
Paper list for equivariant neural network
Protein Ligand INteraction Dataset and Evaluation Resource
Official repository for the Boltz biomolecular interaction models
Plausibility checks for generated molecule poses.
Deep learning framework to train, deploy, and ship AI products Lightning fast.
An open-source PyTorch implementation of the Neighborhood Averaging (NeiA) GNN layer.
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more
OpenMM is a toolkit for molecular simulation using high performance GPU code.
Official implementation of the paper: "ZClip: Adaptive Spike Mitigation for LLM Pre-Training".
Evolutionary Scale Modeling (esm): Pretrained language models for proteins
Universal Structure Alignment of Monomeric and Complex Structure of Nucleic Acids and Proteins
Source code for RNA-FrameFlow: SE(3) Flow Matching for 3D RNA Backbone Design
An implementation of the Equivariant Graph Neural Network (EGNN) layer type for DGL-PyTorch.