Distributed (Deep) Machine Learning Community (@dmlc)

Top repositories

1

xgboost

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
C++
25,402
star
2

dgl

Python package built to ease deep learning on graph, on top of existing DL frameworks.
Python
12,966
star
3

gluon-cv

Gluon CV Toolkit
Python
5,749
star
4

gluon-nlp

NLP made easy
Python
2,548
star
5

nnvm

C++
1,655
star
6

decord

An efficient video loader for deep learning with smart shuffling that's super easy to digest
C++
1,567
star
7

ps-lite

A lightweight parameter server interface
C++
1,502
star
8

minpy

NumPy interface with mixed backend execution
Python
1,111
star
9

mshadow

Matrix Shadow:Lightweight CPU/GPU Matrix and Tensor Template Library in C++/CUDA for (Deep) Machine Learning
C++
1,098
star
10

cxxnet

move forward to https://github.com/dmlc/mxnet
C++
1,025
star
11

dmlc-core

A common bricks library for building scalable and portable distributed machine learning.
C++
861
star
12

dlpack

common in-memory tensor structure
Python
829
star
13

treelite

Universal model exchange and serialization format for decision tree forests
C++
703
star
14

minerva

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++
689
star
15

parameter_server

moved to https://github.com/dmlc/ps-lite
C++
645
star
16

mxnet-notebooks

Notebooks for MXNet
Jupyter Notebook
613
star
17

rabit

Reliable Allreduce and Broadcast Interface for distributed machine learning
C++
507
star
18

mxnet.js

MXNetJS: Javascript Package for Deep Learning in Browser (without server)
JavaScript
435
star
19

MXNet.jl

MXNet Julia Package - flexible and efficient deep learning in Julia
372
star
20

tensorboard

Standalone TensorBoard for visualizing in deep learning
Python
370
star
21

wormhole

Deprecated
C++
341
star
22

mxnet-memonger

Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets
Python
308
star
23

difacto

Distributed Factorization Machines
C++
296
star
24

XGBoost.jl

XGBoost Julia Package
Julia
280
star
25

mxnet-model-gallery

Pre-trained Models of DMLC Project
266
star
26

GNNLens2

Visualization tool for Graph Neural Networks
TypeScript
206
star
27

HalideIR

Symbolic Expression and Statement Module for new DSLs
C++
202
star
28

mxnet-gtc-tutorial

MXNet Tutorial for NVidia GTC 2016.
Jupyter Notebook
131
star
29

experimental-lda

C++
127
star
30

MXNet.cpp

C++ interface for mxnet
C++
114
star
31

experimental-mf

cache-friendly multithread matrix factorization
C++
86
star
32

web-data

The repo to host all the web data including images for documents in dmlc projects.
Jupyter Notebook
80
star
33

nnvm-fusion

Kernel Fusion and Runtime Compilation Based on NNVM
C++
64
star
34

dmlc.github.io

HTML
27
star
35

cub

Cuda
18
star
36

tl2cgen

TL2cgen (TreeLite 2 C GENerator) is a model compiler for decision tree models
C++
13
star
37

mxnet-deepmark

Benchmark speed and other issues internally, before push to deep-mark
Python
7
star
38

mxnet-examples

MXNet Example
6
star
39

xgboost-bench

Python
4
star
40

drat

Drat Repository for DMLC R packages
4
star
41

nn-examples

1
star
42

gluon-nlp-notebooks

1
star
43

docs-redirect-for-mxnet

redirect mxnet.readthedocs.io to mxnet.io
Python
1
star