ptrendx

mxnet

Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more

CoMD-CUDA

GPU implementation of classical molecular dynamics proxy application.

cub

DALI

A library containing both highly optimized building blocks and an execution engine for data pre-processing in deep learning applications

darkness

Dark Themes for Popular Websites

dmlc-core

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

gluon-cv

Gluon CV Toolkit

gluon-nlp

NLP made easy

kokkos

Core repository for Kokkos software

mshadow

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

neomake

A plugin for asynchronous :make using Neovim's job-control functionality

nnvm

Intermediate Computational Graph Representation for Deep Learning Systems

tensorflow

Computation using data flow graphs for scalable machine learning

testing_ground

TransformerEngine

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper GPUs, to provide better performance with lower memory utilization in both training and inference.

tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators

web-data

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