dmlc/wormhole

This repository has been archived on 25/May/2018
Stars
338
Rank 124,931 (Top 3 %)
Language
C++
License
Other
Created over 9 years ago
Updated almost 8 years ago

dmlc/wormhole

dmlc

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Deprecated

This repo is deprecated. We have realized that it is much better to build individual projects that have deep optimization and integration with the existing system.

Pointers to the Projects

XGBoost for tree boosting.
MXNet for for deep learning
DiFaco for factorization.

List of Tools

Boosted Trees (GBDT): XGBoost: eXtreme Gradient Boosting
Clustering: kmeans
Linear method: Asynchrouns SGD L-BFGS
Factorization Machine: DiFacto

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

dgl

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

gluon-cv

Gluon CV Toolkit

gluon-nlp

decord

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

nnvm

ps-lite

A lightweight parameter server interface

minpy

NumPy interface with mixed backend execution

mshadow

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

cxxnet

move forward to https://github.com/dmlc/mxnet

dlpack

common in-memory tensor structure

dmlc-core

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

treelite

Universal model exchange and serialization format for decision tree forests

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.

parameter_server

moved to https://github.com/dmlc/ps-lite

mxnet-notebooks

Notebooks for MXNet

Jupyter Notebook

rabit

Reliable Allreduce and Broadcast Interface for distributed machine learning

mxnet.js

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

MXNet.jl

MXNet Julia Package - flexible and efficient deep learning in Julia

tensorboard

Standalone TensorBoard for visualizing in deep learning

mxnet-memonger

Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets

difacto

Distributed Factorization Machines

XGBoost.jl

XGBoost Julia Package

mxnet-model-gallery

Pre-trained Models of DMLC Project

GNNLens2

Visualization tool for Graph Neural Networks

HalideIR

Symbolic Expression and Statement Module for new DSLs

mxnet-gtc-tutorial

MXNet Tutorial for NVidia GTC 2016.

Jupyter Notebook

experimental-lda

MXNet.cpp

C++ interface for mxnet

experimental-mf

cache-friendly multithread matrix factorization

web-data

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

Jupyter Notebook

nnvm-fusion

Kernel Fusion and Runtime Compilation Based on NNVM

dmlc.github.io

tl2cgen

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

cub

mxnet-deepmark

Benchmark speed and other issues internally, before push to deep-mark

mxnet-examples

xgboost-bench

drat

Drat Repository for DMLC R packages

nn-examples

gluon-nlp-notebooks

docs-redirect-for-mxnet

redirect mxnet.readthedocs.io to mxnet.io