The code for "AttentionPredictor: Temporal Pattern Matters for Efficient LLM Inference", Qingyue Yang, Jie Wang, Xing Li, Zhihai Wang, Chen Chen, Lei Chen, Xianzhi Yu, Wulong Liu, Jianye HAO, Mingx…

This is the code of our paper "Accurate KV Cache Eviction via Anchor Direction Projection for Efficient LLM Inference", which has been accepted by NeurIPS 2025.

Course Project Skeleton for UCAS

This is the source code of our ICML25 paper, titled "Accelerating Large Language Model Reasoning via Speculative Search".

This is the code for our ICLR 2025 paper, titled Computing Circuits Optimization via Model-Based Circuit Genetic Evolution.

This is the code for our paper "Reinforcement Learning within Tree Search for Fast Macro Placement".

This is the code of DiffILO, an unsupervised learning approach for predicting solutions to Integer Linear Programs (ILPs).

Code for our paper: ACM-MILP: Adaptive Constraint Modification via Grouping and Selection for Hardness-Preserving MILP Instance Generation

Official implementation of NeurIPS'24 paper "Reinforcement Learning Policy as Macro Regulator Rather than Macro Placer".

OpenR: An Open Source Framework for Advanced Reasoning with Large Language Models

ChiPBench:Benchmarking End-to-End Performance of AI-based Chip Placement Algorithms

Concise pytorch implements of DRL algorithms, including REINFORCE, A2C, DQN, PPO(discrete and continuous), DDPG, TD3, SAC.

This is the code for G2MILP, a deep learning-based mixed-integer linear programming (MILP) instance generator.

This is the code of paper "A Deep Instance Generative Framework for MILP Solvers Under Limited Data Availability". Zijie Geng, Xijun Li, Jie Wang, Xiao Li, Yongdong Zhang, Feng Wu. NeurIPS 2023 (Sp…

This is the code of paper "A Deep Instance Generative Framework for MILP Solvers Under Limited Data Availability". Zijie Geng, Xijun Li, Jie Wang, Xiao Li, Yongdong Zhang, Feng Wu. NeurIPS 2023 (Sp…

My personal website.

This is the code of paper "De Novo Molecular Generation via Connection-aware Motif Mining". Zijie Geng, Shufang Xie, Yingce Xia, Lijun Wu, Tao Qin, Jie Wang, Yongdong Zhang, Feng Wu, Tie-Yan Liu. I…

This repo awesome-AI4EDA contains the source for the webpage: https://ai4eda.github.io, which is a curated paper list of awesome AI for EDA.

List of Molecular and Material design using Generative AI and Deep Learning

The code of paper Learning Cut Selection for Mixed-Integer Linear Programming via Hierarchical Sequence Model. Zhihai Wang, Xijun Li, Jie Wang*, Yufei Kuang, Mingxuan Yuan, Jia Zeng, Yongdong Zhan…

This is the code of paper "De Novo Molecular Generation via Connection-aware Motif Mining". Zijie Geng, Shufang Xie, Yingce Xia, Lijun Wu, Tao Qin, Jie Wang, Yongdong Zhang, Feng Wu, Tie-Yan Liu. I…

The code of paper LMC: Fast Training of GNNs via Subgraph Sampling with Provable Convergence. Zhihao Shi, Xize Liang, Jie Wang. ICLR 2023.

A novel template-free retrosynthesizer that can generate diverse sets of reactants for a desired product via discrete conditional variational autoencoders.

Results and data for reinforcement learning algorithms.