Jake Lee's repositories
deep-learning-containers
AWS Deep Learning Containers (DLCs) are a set of Docker images for training and serving models in TensorFlow, TensorFlow 2, PyTorch, and MXNet.
Megatron-LM
Ongoing research training transformer models at scale
incubator-mxnet
Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more
minGPT
A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training
mxnet-model-server
Model Server for Apache MXNet is a tool for serving neural net models for inference
mxnet-the-straight-dope
An interactive book on deep learning. Much easy, so MXNet. Wow.
nccl
Optimized primitives for collective multi-GPU communication
NeMo
NeMo: a toolkit for conversational AI
neo-ai-dlr
Neo-AI-DLR is a common runtime for machine learning models compiled by AWS SageMaker Neo, TVM, or TreeLite.
sagemaker-python-sdk
A library for training and deploying machine learning models on Amazon SageMaker
sagemaker-training-toolkit
Train machine learning models within a 🐳 Docker container using 🧠 Amazon SageMaker.
SMDDP-Examples
Repository for Forked Deep Learning Examples Used in SageMaker Data Distributed Training
tensorflow
An Open Source Machine Learning Framework for Everyone