Skip to content

@PKU-DAIR DAIR Lab

Change the repository type filter

All

    Repositories list

    36 repositories

    • Hetu-Galvatron

      Public
      Galvatron is an automatic distributed training system designed for Transformer models, including Large Language Models (LLMs).
      Python
      •
      Apache License 2.0
      •8•78•1•1•Updated Jan 16, 2025Jan 16, 2025

    • Hetu

      Public
      A high-performance distributed deep learning system targeting large-scale and automated distributed training.
      pythondistributed-systemsdata-sciencemachine-learningdeep-neural-networksdeep-learninggpuautogradembeddingsartificial-intelligence
      Python
      •
      Apache License 2.0
      •49•279•1•0•Updated Jan 10, 2025Jan 10, 2025

    • FreeHGC

      Public
      Python
      •1•1•0•0•Updated Dec 6, 2024Dec 6, 2024

    • Starter-Guide

      Public
      A comprehensive guide for beginners in the field of data management and artificial intelligence.
      8•137•1•0•Updated Nov 10, 2024Nov 10, 2024

    • LoRA_Inlaid

      Public
      1•0•0•0•Updated Oct 23, 2024Oct 23, 2024

    • open-box

      Public
      Generalized and Efficient Blackbox Optimization System
      distributed-systemssaasconstrained-optimizationmulti-objective-optimizationbayesian-optimizationhyper-parameter-optimizationblackbox-optimizationautomatic-machine-learning
      Python
      •
      Other
      •81•392•11•1•Updated Oct 17, 2024Oct 17, 2024

    • RAG-Survey

      Public
      Collecting awesome papers of RAG for AIGC. We propose a taxonomy of RAG foundations, enhancements, and applications in paper "Retrieval-Augmented Generation for AI-Generated Content: A Survey".
      94•251•0•0•Updated Jun 21, 2024Jun 21, 2024

    • SGL

      Public
      A scalable graph learning toolkit for extremely large graph datasets. (WWW'22, 🏆 Best Student Paper Award)
      deep-learninggeometric-deep-learningscalable-graph-learningpytorchnetwork-embeddingneural-architecture-searchgraph-neural-networks
      Python
      •
      MIT License
      •22•147•0•2•Updated May 10, 2024May 10, 2024

    • CAFE

      Public
      [SIGMOD 2024] CAFE: Towards Compact, Adaptive, and Fast Embedding for Large-scale Recommendation Models
      Python
      •5•2•0•0•Updated Dec 7, 2023Dec 7, 2023

    • OUTRE

      Public
      OUTRE: An OUT-of-core De-REdundancy GNN Training Framework for Massive Graphs within A Single Machine
      C++
      •1•2•0•0•Updated Oct 21, 2023Oct 21, 2023

    • MEVI

      Public
      [NeurIPS 2023] Model-enhanced Vector Index
      Python
      •3•1•0•0•Updated Sep 26, 2023Sep 26, 2023

    • DBTune

      Public
      A customized and efficient database tuning system [VLDB'22]
      Python
      •
      Other
      •21•33•1•0•Updated Sep 21, 2023Sep 21, 2023

    • HGAMLP

      Public
      HGAMLP: A Scalable Training Framework for Heterogeneous Graph Neural Networks
      Python
      •0•5•1•0•Updated Jul 15, 2023Jul 15, 2023

    • AIR

      Public
      Python
      •2•7•0•0•Updated Sep 26, 2022Sep 26, 2022

    • SSD

      Public
      Transfer Learning based Search Space Design for Hyperparameter Tuning [SIGKDD'22]
      Python
      •1•4•0•0•Updated Aug 25, 2022Aug 25, 2022

    • GNN-in-RS

      Public
      GNN in Recommendation System (ACM Computing Surveys 2022)
      65•83•0•0•Updated Jun 25, 2022Jun 25, 2022

    • GAMLP

      Public
      Code of GAMLP for Open Graph Benchmark. KDD‘22
      Python
      •13•61•0•0•Updated Jun 25, 2022Jun 25, 2022

    • NAFS

      Public
      ICML‘22 NAFS
      Python
      •5•5•0•0•Updated Jun 20, 2022Jun 20, 2022

    • KnobsTuningEA

      Public
      An Experimental Evaluation for Database Configuration Tuning
      ReScript
      •15•23•2•0•Updated Mar 15, 2022Mar 15, 2022

    • DFG-NAS

      Public
      ICML‘22 DF-GNAS
      Python
      •2•1•0•0•Updated Jan 28, 2022Jan 28, 2022

    • 2021_CCF_BDCI_LargeBERT_Rank1st

      Public
      Cuda
      •
      Apache License 2.0
      •0•8•0•0•Updated Jan 20, 2022Jan 20, 2022

    • 2021_AIAC_Task2_1st

      Public
      QQ浏览器2021AI算法大赛赛道二 第1名 方案. Rank 1st solution to QQ Browser 2021 AI Algorithm Competition (CIKM AnalytiCup 2021) Track 2 Automated Hyperparameter Optimization.
      Python
      •6•26•0•0•Updated Dec 7, 2021Dec 7, 2021

    • HyperTune

      Public
      Efficient Hyper-parameter Tuning at Scale (VLDB'22)
      Python
      •3•8•0•0•Updated Dec 1, 2021Dec 1, 2021

    • mindware

      Public
      An efficient open-source AutoML system for automating machine learning lifecycle, including feature engineering, neural architecture search, and hyper-parameter tuning.
      pythondistributed-systemsdata-sciencemachine-learningdeep-learningensemble-learningbayesian-optimizationneural-architecture-searchmeta-learninghyper-parameter-optimization
      Python
      •
      MIT License
      •27•55•0•0•Updated Nov 16, 2021Nov 16, 2021

    • RIM

      Public
      RIM: Reliable Influence-based Active Learning on Graphs (NeurIPS'21 Spotlight)
      2•1•0•0•Updated Oct 27, 2021Oct 27, 2021

    • NDLS

      Public
      Node Dependent Local Smoothing for Scalable Graph Learning(NeurIPS'21, Spotlight)
      Python
      •1•0•0•0•Updated Oct 27, 2021Oct 27, 2021

    • IGP

      Public
      Active Learning for GNN (ICLR'22)
      5•5•0•0•Updated Oct 6, 2021Oct 6, 2021

    • MFES-HB

      Public
      Implementation of MFES-HB [AAAI'21] along with Hyperband and BOHB
      Python
      •
      MIT License
      •1•5•0•0•Updated Sep 28, 2021Sep 28, 2021

    • ROD

      Public
      ROD: Reception-aware Online Distillation for Sparse Graphs [KDD 2021]
      3•0•0•0•Updated Sep 14, 2021Sep 14, 2021

    • nnfusion

      Public
      A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description. [OSDI 2020]
      C++
      •
      MIT License
      •161•1•0•0•Updated Sep 10, 2021Sep 10, 2021