Repositories list

• xgboost

  Public
  Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on single machine, Hadoop, Spark, Dask, Flink and DataFlow

  distributed-systemsxgboostgbdt+ 3gbmgbrtmachine-learning

  C++

  •

  Apache License 2.0

  •8.7k•26k•400•60•Updated Dec 28, 2024Dec 28, 2024

• xgboost-devops

  Public
  Host custom actions; keep track of manual approval requests for CI jobs.

  Shell

  •

  Apache License 2.0

  •1•4•0•0•Updated Dec 10, 2024Dec 10, 2024

• XGBoost.jl

  Public
  XGBoost Julia Package

  Julia

  •

  Other

  •109•289•24•5•Updated Nov 27, 2024Nov 27, 2024

• gluon-cv

  Public
  Gluon CV Toolkit

  machine-learningcomputer-visionneural-network+ 10ganimage-classificationobject-detectionsemantic-segmentationaction-recognitionpose-estimationperson-reid+ 3

  Python

  •

  Apache License 2.0

  •1.2k•5.9k•40•21•Updated Nov 25, 2024Nov 25, 2024

• treelite

  Public
  Universal model exchange and serialization format for decision tree forests

  C++

  •

  Apache License 2.0

  •101•746•9•0•Updated Nov 22, 2024Nov 22, 2024

• dgl

  Public
  Python package built to ease deep learning on graph, on top of existing DL frameworks.

  deep-learninggraph-neural-networks

  Python

  •

  Apache License 2.0

  •3k•14k•424•127•Updated Oct 18, 2024Oct 18, 2024

• tl2cgen

  Public
  TL2cgen (TreeLite 2 C GENerator) is a model compiler for decision tree models

  C++

  •

  Apache License 2.0

  •6•24•21•2•Updated Oct 3, 2024Oct 3, 2024

• dlpack

  Public
  common in-memory tensor structure

  deep-learningoperatortensor

  Python

  •

  Apache License 2.0

  •133•924•30•3•Updated Sep 28, 2024Sep 28, 2024

• decord

  Public
  An efficient video loader for deep learning with smart shuffling that's super easy to digest

  video-loader

  C++

  •

  Apache License 2.0

  •167•1.9k•180•8•Updated Jul 17, 2024Jul 17, 2024

• dmlc-core

  Public
  A common bricks library for building scalable and portable distributed machine learning.

  dmlc

  C++

  •

  Apache License 2.0

  •519•869•32•0•Updated Jun 14, 2024Jun 14, 2024

• dmlc.github.io

  Public
  HTML

  •38•27•1•1•Updated Jun 4, 2024Jun 4, 2024

• gluon-nlp

  Public archive
  NLP made easy

  nlpmachine-learningnatural-language-processing+ 10deep-learningmxnetnumpynlugluonnatural-language-generationnlg+ 3

  Python

  •

  Apache License 2.0

  •533•2.6k•237•35•Updated Oct 6, 2023Oct 6, 2023

• DeepSpeed

  Public
  DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

  Python

  •

  Apache License 2.0

  •4.2k•4•0•0•Updated Jul 27, 2023Jul 27, 2023

• ps-lite

  Public
  A lightweight parameter server interface

  C++

  •

  Apache License 2.0

  •542•1.5k•59•14•Updated Jan 11, 2023Jan 11, 2023

• GNNLens2

  Public
  Visualization tool for Graph Neural Networks

  visualizationdeep-learningpytorch+ 6graph-visualizationxaigraph-neural-networksgraph-representation-learningexplainabilitydgl

  TypeScript

  •

  Apache License 2.0

  •27•242•5•0•Updated Sep 20, 2022Sep 20, 2022

• web-data

  Public
  The repo to host all the web data including images for documents in dmlc projects.

  Jupyter Notebook

  •

  Apache License 2.0

  •154•84•2•2•Updated Jul 25, 2022Jul 25, 2022

• mxnet-model-gallery

  Public
  Pre-trained Models of DMLC Project

  Other

  •74•266•23•2•Updated Apr 15, 2021Apr 15, 2021

• rabit

  Public
  Reliable Allreduce and Broadcast Interface for distributed machine learning

  C++

  •

  BSD 3-Clause "New" or "Revised" License

  •184•511•0•0•Updated Nov 5, 2020Nov 5, 2020

• mxnet.js

  Public
  MXNetJS: Javascript Package for Deep Learning in Browser (without server)

  JavaScript

  •

  Apache License 2.0

  •73•435•7•1•Updated Oct 11, 2020Oct 11, 2020

• HalideIR

  Public
  Symbolic Expression and Statement Module for new DSLs

  C++

  •

  Other

  •59•205•2•0•Updated Oct 6, 2020Oct 6, 2020

• gluon-nlp-notebooks

  Public
  3•1•0•0•Updated Jun 29, 2020Jun 29, 2020

• tensorboard

  Public archive
  Standalone TensorBoard for visualizing in deep learning

  visualizationmxnettensorboard

  Python

  •

  Apache License 2.0

  •60•371•16•4•Updated Mar 24, 2020Mar 24, 2020

• ccache

  Public
  ccache – a fast compiler cache

  C++

  •

  Other

  •506•2•0•0•Updated Mar 16, 2020Mar 16, 2020

• xgboost-bench

  Public
  Python

  •4•4•3•0•Updated Dec 19, 2019Dec 19, 2019

• numpy-ml

  Public
  Machine learning, in numpy

  Python

  •

  GNU General Public License v3.0

  •3.8k•5•0•0•Updated Sep 11, 2019Sep 11, 2019

• mshadow

  Public archive
  Matrix Shadow:Lightweight CPU/GPU Matrix and Tensor Template Library in C++/CUDA for (Deep) Machine Learning

  C++

  •

  Other

  •432•1.1k•37•0•Updated Aug 4, 2019Aug 4, 2019

• MXNet.jl

  Public
  MXNet Julia Package - flexible and efficient deep learning in Julia

  70•370•40•5•Updated Jul 19, 2019Jul 19, 2019

• minerva

  Public archive
  Minerva: a fast and flexible tool for deep learning on multi-GPU. It provides ndarray programming interface, just like Numpy. Python bindings and C++ bindings are both available. The resulting code can be run on CPU or GPU. Multi-GPU support is very easy.

  C++

  •

  Other

  •172•701•9•2•Updated Nov 1, 2018Nov 1, 2018

• nnvm

  Public archive
  deep-learningdeploymentmetal+ 7optimizationopenclcudacomputation-graphrocmtvmnnvm

  C++

  •

  Apache License 2.0

  •281•1.7k•1•0•Updated Sep 11, 2018Sep 11, 2018

• keras

  Public
  Deep Learning library for Python. Convnets, recurrent neural networks, and more. Runs on MXNet, Theano or TensorFlow.

  Python

  •

  Other

  •19k•125•18•4•Updated Sep 6, 2018Sep 6, 2018