Code for BLT research paper

This is the repository that contains the source code for the Self-Evaluation Guided MCTS for online DPO.

Reference implementation for Token-level Direct Preference Optimization(TDPO)

Structure-based drug design based on Retrieval Augmented Generation

Generative modeling of molecular dynamics trajectories

Artificial Intelligence Research for Science (AIRS)

List of papers about Peptide research using Deep Learning

This is the official repository of Patch aggregation for Graph Neural Networks.

AI-powered ab initio biomolecular dynamics simulation

A distance distribution-based visualization method for atomic pair energy and precise prediction of absolute binding free energy.

Token-Mol 1.0：tokenized drug design with large language model

MedicalGPT: Training Your Own Medical GPT Model with ChatGPT Training Pipeline. 训练医疗大模型，实现了包括增量预训练(PT)、有监督微调(SFT)、RLHF、DPO、ORPO。

This repository showcases various advanced techniques for Retrieval-Augmented Generation (RAG) systems. RAG systems combine information retrieval with generative models to provide accurate and cont…

code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"

We present the first systematic study on the scaling property of raw agents instantiated by LLMs. We find that performance scales with the increase in the number of agents, using the simple(st) way…

Code for the tutorial/review paper for RL-based-fine-tuniing. In this code, we especially focus on the design of biological sequences like DNA (enhancers) and RNA (UTRs) design.

[ICLR 2024] Protein-Ligand Interaction Prior for Binding-aware 3D Molecule Diffusion Models

Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"

Unified Efficient Fine-Tuning of 100+ LLMs (ACL 2024)

Diffusion-based all-atom protein generative model.

Accelerating Geometric Latent Diffusion Models for 3D Molecular Conformation Generation

Geometric Latent Diffusion Models for 3D Molecule Generation

EC-Conf: An Ultra-fast Diffusion Model for Molecular conformation Generation with Equivariant Consistency

FedGT: Federated Node Classification with Scalable Graph Transformer

DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support

Reference implementation for DPO (Direct Preference Optimization)