One Transformer Can Understand Both 2D & 3D Molecular Data. - Shengjie Luo; Tianlang Chen 001; Yixian Xu; Shuxin Zheng; Tie-Yan Liu; Liwei Wang 001; Di He 001 (ICLR, 2023)

Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Products. - Shengjie Luo; Tianlang Chen 001; Aditi S. Krishnapriyan (ICLR, 2024)

GeoMFormer: A General Architecture for Geometric Molecular Representation Learning. - Tianlang Chen 001; Shengjie Luo; Di He 001; Shuxin Zheng; Tie-Yan Liu; Liwei Wang 001 (ICML, 2024)

RelGNN: Composite Message Passing for Relational Deep Learning. - Tianlang Chen 001; Charilaos I. Kanatsoulis; Jure Leskovec (ICML, 2025)

Exchangeability in Neural Network Architectures and its Application to Dynamic Pruning. - Pu Yi 001; Tianlang Chen 001; Yifan Yang; Sara Achour (CoRR, 2025)

Fractional Reasoning via Latent Steering Vectors Improves Inference Time Compute. - Sheng Liu; Tianlang Chen 001; Pan Lu; Haotian Ye; Yizheng Chen; Lei Xing 001; James Zou 001 (CoRR, 2025)

RFG: Test-Time Scaling for Diffusion Large Language Model Reasoning with Reward-Free Guidance. - Tianlang Chen 001; Minkai Xu; Jure Leskovec; Stefano Ermon (CoRR, 2025)

Improving Diffusion Language Model Decoding through Joint Search in Generation Order and Token Space. - Yangyi Shen; Tianjian Feng; Jiaqi Han; Wen Wang 015; Tianlang Chen 001; Chunhua Shen; Jure Leskovec; Stefano Ermon (CoRR, 2026)

Found in Conversation: LLMs Teach Themselves to Close the Multi-Turn Gap. - Tianlang Chen 001; Shirley Wu; Jure Leskovec (CoRR, 2026)