aws-samples / machine-learning-using-k8s

Train and Deploy Machine Learning Models on Kubernetes using Amazon EKS

Machine Learning Frameworks on Kubernetes

This repository explains how to run different Machine Learning frameworks on Amazon EKS.

• Create GPU-enabled Amazon EKS cluster with Kubeflow
• Single node training for MNIST
  • TensorFlow and Keras
  • MXNet and Keras
• Inference for MNIST
  • TensorFlow Serving Component
  • MXNet Model Server
• TensorBoard setup
  • TensorBoard on Amazon EKS
• Distributed training
  • TensorFlow and Horovod on Amazon EKS with ImageNet Data
  • TensorFlow and Horovod on Amazon EKS with Synthetic Data
  • MNIST Handwritten Digit Recognition in PyTorch

Other Notes

• Jupyter Notebook on Amazon EKS
• Serverless Inference
  • Serverless inference on AWS Fargate
• Inference using Imagenet
  • TensorFlow on Amazon EKS