Stars
[ICLR 2026] On the Generalization of SFT: A Reinforcement Learning Perspective with Reward Rectification.
IMO-2025 SOLUTION FROM HUAWEI XIAOYI AI TEAM
The official implementation for ICML24 paper "Graph Out-of-Distribution Detection Goes Neighborhood Shaping"
The official implementation for ICLR25 paper "Beyond Circuit Connections: A Non-Message Passing Graph Transformer Approach Towards Quantum Error Mitigation"
Official implementation for the paper "HShare: Fast LLM Decoding by Hierarchical Key-Value Sharing" (ICLR 2025).
Official Implementation of "Drug-drug interaction prediction via hierarchical structure modeling" (SCIENTIA SINICA Informationis) "基于层次化结构建模的药物相互作用类型预测" (中国科学信息科学)
16-fold memory access reduction with nearly no loss
Official implementation for the paper "QVAE-Mole: The Quantum VAE with Spherical Latent Variable Learning for 3-D Molecule Generation" (NeurIPS 2024).
Reasoning 3D Segmentation - "segment anything"/grounding/part seperation in 3D with natural conversations.
Official implementation for the paper "EBMDOCK: NEURAL PROBABILISTIC PROTEIN-PROTEIN DOCKING VIA A DIFFERENTIABLE ENERGY-BASED MODEL" (ICLR 2024).
We would like to maintain a list of resources which aim to solve molecular docking and other closely related tasks.
The official implementation of NeurIPS22 spotlight paper "NodeFormer: A Scalable Graph Structure Learning Transformer for Node Classification"
OOD Dataset Curator and Benchmark for AI-aided Drug Discovery
This is a tensorflow-based rotation detection benchmark, also called AlphaRotate.
PyTorch implementation of Image-to-Image Translation with Conditional Adversarial Nets (pix2pix)
CRFL: Certifiably Robust Federated Learning against Backdoor Attacks (ICML 2021)
Federated learning on graph, especially on graph neural networks (GNNs), knowledge graph, and private GNN.
zjtone / keccak-python
Forked from chris-wood/keccak-pythonSHA-3 Keccak sponge function implemented in Python