machine-learning-using-k8s icon indicating copy to clipboard operation
machine-learning-using-k8s copied to clipboard

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