young768 Goto Github PK

adaptdl

Resource-adaptive cluster scheduler for deep learning training.

batch_adaptive

Straggler_

bigbird

Transformers for Longer Sequences

byteps

A high performance and generic framework for distributed DNN training

code_pool

cuda-samples

Samples for CUDA Developers which demonstrates features in CUDA Toolkit

cuda-winograd

Fast CUDA Kernels for ResNet Inference.

cudalibrarysamples

CUDA Library Samples

cudnn-sample

cuDNN sample codes provided by Nvidia

deeplearningexamples

Deep Learning Examples

deeprecsys

http://vlsiarch.eecs.harvard.edu/research/recommendation/

deepspeed

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

demo

dtr-prototype

Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616

dyang

fairscale

PyTorch extensions for high performance and large scale training.

google-research

Google Research

graph_project_start

iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.

iree_script

ispark

Elastic Executor Provisioning for Iterative Workloads on Apache Spark

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

jax-test

jax_custom_ops_and_custom_partitioning

kungfu

KungFu: Making Training in Distributed Machine Learning Adaptive

leetcode-solutions

:octocat: (Weekly Update) Python / C++11 Solutions of All 1263 LeetCode Problems

megatron-lm

Ongoing research training transformer models at scale

mixture-of-experts

A Pytorch implementation of Sparsely-Gated Mixture of Experts, for massively increasing the parameter count of language models

mlir-tutorial

movies-recommendation