HabanaAI/Gaudi-solutions

Full End-to-End examples showing how to use First-gen Gaudi and Gaudi2 in common use cases

Gaudi Solutions

This repository contains Jupiter notebooks with step-by-step instructions for training deep learning models on Gaudi for different industry verticals.

The notebooks run on a Gaudi system with SynapseAI and Gaudi drivers installed. Please refer to installation guide for more information.

Tutorials will cover training, evaluation, and inferencing for the following models:

• Interactive Chat
  • Chat using the Llama 2 Large Language Model
• Healthcare
  • CheXNet: Classification of Thoracic Diseases From Chest X-ray Images
• FinTech
  • FinBERT: Financial Sentiment Analysis using Language Models
• Retail
  • Retail: Fine Tuning Image Detection for Retail Items using YOLOX
• Autonomous Driving
  • Autonomous Driving: Semantic Segmentation using PyTorch UNet 2D