THUDM

slime is an LLM training framework for RL Scaling.

A Comprehensive Benchmark to Evaluate LLMs as Agents (ICLR'24)

GLM (General Language Model)

An optimized deep prompt tuning strategy comparable to fine-tuning across scales and tasks

[ICLR 2025] LongWriter: Unleashing 10,000+ Word Generation from Long Context LLMs

CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)

WebGLM: An Efficient Web-enhanced Question Answering System (KDD 2023)

AgentTuning: Enabling Generalized Agent Abilities for LLMs

LongBench v2 and LongBench (ACL 25'&24')

SwissArmyTransformer is a flexible and powerful library to develop your own Transformer variants.

A novel method to tune language models. Codes and datasets for paper ``GPT understands, too''.

An LLM-based Web Navigating Agent (KDD'24)

ReST-MCTS*: LLM Self-Training via Process Reward Guided Tree Search (NeurIPS 2024)

GraphMAE: Self-Supervised Masked Graph Autoencoders in KDD'22

INFTY Engine: An Optimization Toolkit to Support Continual AI

Building Open LLM Web Agents with Self-Evolving Online Curriculum RL

Source code and dataset for KDD 2019 paper "Representation Learning for Attributed Multiplex Heterogeneous Network"

LongCite: Enabling LLMs to Generate Fine-grained Citations in Long-context QA

Source code and dataset for ACL 2019 paper "Cognitive Graph for Multi-Hop Reading Comprehension at Scale"

大模型多维度中文对齐评测基准 (ACL 2024)

Scaling Agentic Reinforcement Learning with a Multi-Turn, Multi-Task Framework

DeepDive: Advancing Deep Search Agents with Knowledge Graphs and Multi-Turn RL

Revisiting, benchmarking, and refining Heterogeneous Graph Neural Networks.

GCC: Graph Contrastive Coding for Graph Neural Network Pre-Training @ KDD 2020

Official Pytorch Implementation for MathGLM

Source code and dataset for KDD 2020 paper "Controllable Multi-Interest Framework for Recommendation"

Towards Large Multimodal Models as Visual Foundation Agents

[EMNLP 2024] LongAlign: A Recipe for Long Context Alignment of LLMs

Towards Knowledge-Based Personalized Product Description Generation in E-commerce @ KDD 2019

Source code and dataset for IJCAI 2019 paper "ProNE: Fast and Scalable Network Representation Learning"

Paper list of NLP for recommender systems

Source code and dataset of the NeurIPS 2020 paper "Graph Random Neural Network for Semi-Supervised Learning on Graphs"

GraphMAE2: A Decoding-Enhanced Masked Self-Supervised Graph Learner in WWW'23

A unified tokenization tool for Images, Chinese and English.

IndexCache: Accelerating Sparse Attention via Cross-Layer Index Reuse

Towards Knowledge-Based Recommender Dialog System @ EMNLP 2019

Finetuning and inference tools for the CogView4 and CogVideoX model series.

Protein Language Model

Code, Data and Demo for Paper: Controllable Generation from Pre-trained Language Models via Inverse Prompting

RL Scaling and Test-Time Scaling (ICML'25)

Source code and dataset for KDD 2020 paper "Understanding Negative Sampling in Graph Representation Learning"

Parameter-Efficient Fine-Tuning for Foundation Models

TreeRL: LLM Reinforcement Learning with On-Policy Tree Search in ACL'25

Graph Robustness Benchmark: A scalable, unified, modular, and reproducible benchmark for evaluating the adversarial robustness of Graph Machine Learning.

SciGLM: Training Scientific Language Models with Self-Reflective Instruction Annotation and Tuning (NeurIPS D&B Track 2024)

kgTransformer: pre-training for reasoning over complex KG queries (KDD 22)

Implementation of "GraphSGAN", a GAN-based semi-supervised learning algorithm for graph data.

A heterogeneous entity-augmented academic language model based on Open Academic Graph (OAG)

Source code and dataset for KDD 2019 paper "Sequential Scenario-Specific Meta Learner for Online Recommendation"

Codes for WWW2022 accepted paper: SelfKG: Self-Supervised Entity Alignment in Knowledge Graphs

This repository contains the code and data for the paper "Chaining the Evidence: Robust Reinforcement Learning for Deep Search Agents with Citation-Aware Rubric Rewards".

Open-source framework for computer use agents: VeriGen verifiable task synthesis, online RL training (AgentRL), and OSWorld/ScienceBoard evaluation.

Source code and dataset for paper "Cognitive Knowledge Graph Reasoning for One-shot Relational Learning"

NaturalCodeBench (Findings of ACL 2024)

[ACL25' Findings] SWE-Dev is an SWE agent with a scalable test case construction pipeline.

DataSciBench: An LLM Agent Benchmark for Data Science (Findings of ACL 2026)

The multilingual variant of GLM, a general language model trained with autoregressive blank infilling objective

Source code and dataset for KDD 2019 paper "OAG: Toward Linking Large-scale Heterogeneous Entity Graphs"

KDD'23 Web-Scale Academic Name Disambiguation: the WhoIsWho Benchmark, Leaderboard, and Toolkit