Gaudi - Habana Developers

Models

Yolov5

Yolov5 model for advanced object detection

Learn more

Models

Wide & Deep

Wide & Deep Recommender Model Training For TensorFlow 2

Learn more

Models

Vision Transformer

Vision Transformer – Image classification using transformers

Learn more

Models

VGG-Segnet

VGG-Segnet for TensorFlow

Learn more

Models

UNet3D

UNet3D for TensorFlow

Learn more

Models

UNet3D

UNet3D model for PyTorch

Learn more

Models

Unet2D

UNet for 2D Medical Image Segmentation for TensorFlow

Learn more

Models

UNet2D

UNet2D model for PyTorch

Learn more

Models

UNet Industrial

UNet Industrial Defect Segmentation

Learn more

Models

Transformer

Transformer for TensorFlow

Learn more

Models

Transformer

Transformer model for PyTorch

Learn more

Models

T5

T5 for TensorFlow

Learn more

Models

Swin Transformer

Swin Transformer – Shifted Window Model for Computer Vision

Learn more

Models

SSD ResNet34

Single Shot Detection (SSD) for TensorFlow

Learn more

Models

SSD

SSD model for Object Detection for PyTorch

Learn more

Models

RoBERTa

RoBERTa for PyTorch

Learn more

Models

RetinaNet

RetinaNet for TensorFlow

Learn more

Models

ResNeXt

ResNeXt for PyTorch

Learn more

Models

ResNeXt

ResNeXt for TensorFlow

Learn more

Models

ResNet

ResNet for PyTorch

Learn more

Models

ResNet

ResNet for TensorFlow

Learn more

Models

MobileNetV2

MobileNetV2 for TensorFlow

Learn more

Models

Mobile Net V2

MobileNet v2 classification model for PyTorch

Learn more

Models

Mask R-CNN

Mask R-CNN for TensorFlow

Learn more

Models

GPT-2 small

GPT-2 small model for Natural Language Processing

Learn more

Models

GoogLeNet

GoogLeNet classification model for PyTorch

Learn more

Models

ELECTRA Fine Tuning

ELECTRA Fine Tuning For TensorFlow2

Learn more

Models

ELECTRA Fine Tuning

ELECTRA Large Discriminator model for Fine Tuning

Learn more

Models

EfficientDet

EfficientDet for TensorFlow

Learn more

Models

DistilBERT

DistilBERT – small, fast

Learn more

Models

DistilBERT

DistilBERT model from Huggingface repository

Learn more

Models

DINO

PyTorch implementation for DINO; based on code from facebookresearch/dino repository

Learn more

Models

DenseNet

DenseNetfor TensorFlow

Learn more

Models

CycleGAN

CycleGAN for TensorFlow

Learn more

Models

BERT-1.5B

BERT model with 1.5 Billion Parameters using DeepSpeed for NLP

Learn more

Models

BERT Pretraining

BERT Pretraining for PyTorch

Learn more

Models

BERT Fine tuning

BERT Fine tuning for PyTorch

Learn more

Models

BERT

BERT for TensorFlow

Learn more

Models

BART

BART model for NLP on Pytorch

Learn more

Models

ALBERT

ALBERT for TensorFlow

Learn more

Models

ALBERT

ALBERT model for NLP on PyTorch

Learn more