collaborators

5 papers

q-bio.BM2025

Aligning Protein Conformation Ensemble Generation with Physical Feedback

Jiarui Lu, Xiaoyin Chen, Stephen Zhewen Lu +4

Protein dynamics play a crucial role in protein biological functions and properties, and their traditional study typically relies on time-consuming molecular dynamics (MD) simulati…

cs.CL2025

Can Memory-Augmented Language Models Generalize on Reasoning-in-a-Haystack Tasks?

Payel Das, Ching-Yun Ko, Sihui Dai +3

Large language models often expose their brittleness in reasoning tasks, especially while executing long chains of reasoning over context. We propose MemReasoner, a new and simple…

cs.LG2024

Multi-Scale Representation Learning for Protein Fitness Prediction

Zuobai Zhang, Pascal Notin, Yining Huang +5

Designing novel functional proteins crucially depends on accurately modeling their fitness landscape. Given the limited availability of functional annotations from wet-lab experime…

cs.LG2024

Larimar: Large Language Models with Episodic Memory Control

Payel Das, Subhajit Chaudhury, Elliot Nelson +10

Efficient and accurate updating of knowledge stored in Large Language Models (LLMs) is one of the most pressing research challenges today. This paper presents Larimar - a novel, br…

cs.LG2024

NeuroPrune: A Neuro-inspired Topological Sparse Training Algorithm for Large Language Models

Amit Dhurandhar, Tejaswini Pedapati, Ronny Luss +4

Transformer-based Language Models have become ubiquitous in Natural Language Processing (NLP) due to their impressive performance on various tasks. However, expensive training as w…