Zuobai Zhang's starred repositories
robust-laplacians-py
Build high-quality Laplace matrices on meshes and point clouds in Python. Implements [Sharp & Crane SGP 2020].
equiformer_v2
[ICLR'24] EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree Representations
DynamicBind
repo for DynamicBind: Predicting ligand-specific protein-ligand complex structure with a deep equivariant generative model
discrete_flow_models
Code for the paper https://arxiv.org/abs/2402.04997
scalar-fields
Equivariant Scalar Fields for Molecular Docking with Fast Fourier Transforms
ProteinWorkshop
Benchmarking framework for protein representation learning. Includes a large number of pre-training and downstream task datasets, models and training/task utilities. (ICLR 2024)
ProteinNPT
Official code repository for the paper "ProteinNPT: Improving Protein Property Prediction and Design with Non-Parametric Transformers"
protein-vec
Repository for Protein-Vec, a protein embedding mixture of experts model
Tranception
Official repository for the paper "Tranception: Protein Fitness Prediction with Autoregressive Transformers and Inference-time Retrieval"
ESM-GearNet
ESM-GearNet for Protein Structure Representation Learning (https://arxiv.org/abs/2303.06275)
conditional-flow-matching
TorchCFM: a Conditional Flow Matching library
Structure-informed_PLM
Protein language modeling with information from structures