readme.md

Machine Learning Frameworks on Kubernetes

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

• Create GPU-enabled Amazon EKS cluster
• Setup KubeFlow on Amazon EKS
• Single node training
  • TensorFlow + Keras on Amazon EKS
  • MXNet on Amazon EKS
  • Jupyter notebook for TensorFlow on Amazon EKS
  • TensorFlow on Amazon EKS
• Distributed training
  • TensorFlow and Horovod on Amazon EKS with ImageNet Data
  • TensorFlow and Horovod on Amazon EKS with Synthetic Data
• TensorBoard setup
  • TensorBoard on Amazon EKS
• Inference
  • Image Recognition using TensorFlow on Amazon EKS