IPU-powered Jupyter Notebooks
Build, train and deploy your machine learning models faster using IPU-powered Jupyter notebooks in the cloud with Paperspace. With just one click, access a wide range of ML models and solutions for Computer Vision, Natural Language Processing, Graph Neural Networks, and more.
Notebook
Model(s)
ML Domain
Framework
Type
POD4 Capability
Enhanced POD16 Capability
Llama 2: A Chatbot LLM on IPUs - Inference
Llama 2 7Bn/13Bn
Natural Language Processing
Hugging Face
Inference
Yes (run faster on POD16)
Dolly 2.0 – The World’s First, Truly Open Instruction-Tuned LLM on IPUs – Inference
Dolly 2.0
Natural Language Processing
PyTorch
Inference
Yes (run faster on POD16)
Chatbot on IPUs using OpenAssistant Pythia 12B - Inference
Pythia 12B
Natural Language Processing
Hugging Face
Inference
Yes (run faster on POD16)
Flan-T5 is (probably) all you need - Inference
Flan-T5
Natural Language Processing
Hugging Face
Inference
Yes (POD16 enables Flan-T5-XL)
Textual Entailment on IPUs using Flan-T5 XL & XXL - Fine-Tuning
Flan-T5
Natural Language Processing
Hugging Face
Fine-Tuning
Yes
Text Generation with Flan-T5 XL on Pod4 - Inference
Flan-T5
Natural Language Procesing
Hugging Face
Inference
-
Text Generation with Flan-T5 XXL on Pod16 - Inference
Flan-T5
Natural Language Processing
Hugging Face
Inference
Yes (POD16 enables Flan-T5 XXL)
Machine Translation on IPUs using MT5 - Fine-tuning
MT5-small
Natural Language Processing
Hugging Face
Fine-Tuning
Zero-Shot Text Classification on IPUs using MT5 - Inference
MT5-large
Natural Language Processing
Hugging Face
Inference
Yes (POD16 enables MT5-large)
Text entailment on IPU using GPT-J - Fine-tuning
GPT-J 6B
Natural Language Processing
PyTorch
Fine-Tuning
Yes (POD16 enables fine-tuning GPT-J)
Text generation on IPU using GPT-J - Inference
GPT-J 6B
Natural Language Processing
PyTorch
Inference
Yes (POD16 enables inference GPT-J)
Faster Text Generation with GPT-J using 4-bit Weight Quantization on IPUs
GPT-J-6B
Natural Language Processing
PyTorch
Inference
Yes (run faster on POD16)
Multi-lingual ASR Transcription on IPUs using Whisper - Fine-tuning
Whisper
Speech Processing
Hugging Face
Fine-Tuning
Yes (run faster on POD16)
Speech Transcription on IPUs using Whisper - Inference
Whisper
Speech Processing
Hugging Face
Inference
Yes (POD16 enables Whisper-Large)
Speech Transcription on IPUs using Whisper - Inference using Group Quantisation
Whisper
Speech Processing
Hugging Face
Inference
Yes (run faster on POD16)
Text-to-Image Generation using Stable Diffusion 2 on IPU
Stable Diffusion 2
Computer Vision
Hugging Face
Inference
-
Image-to-Image Generation on IPU using Stable Diffusion - Inference
Stable Diffusion
Computer Vision
Hugging Face
Inference
-
Text-to-Image Generation on IPU with Stable Diffusion - Inference
Stable Diffusion
Computer Vision
Hugging Face
Inference
-
Text Guided In-Painting on IPU using Stable Diffusion - Inference
Stable Diffusion
Computer Vision
Hugging Face
Inference
-
Object detection on IPU using YOLO v4 - inference
YOLO v4
Computer Vision
PyTorch
Inference
-
Multi-label classification on IPU using ViT - Fine-Tuning
ViT
Computer Vision
PyTorch (on Hugging Face model)
Fine-Tuning
Yes (run faster on POD16)
Image Classification on IPU with ViT - Fine-Tuning
ViT
Computer Vision
Hugging Face
Fine-Tuning
Yes (run faster on POD16)
Faster single-label text classification on IPU with Packed BERT - Fine-Tuning & Inference
Packed BERT
Natural Language Processing
Hugging Face
Fine-Tuning
Yes (run faster on POD16)
Faster multi-label text classification on IPU with Packed BERT - Fine-Tuning & Inference
Packed BERT
Natural Language Processing
Hugging Face
Fine-Tuning
Yes (run faster on POD16)
Faster question-answering on IPU using Packed BERT - Fine-Tuning & Inference
Packed BERT
Natural Language Processing
Hugging Face
Fine-Tuning
Yes (run faster on POD16)
Question-Answering on IPU using BERT - Fine-Tuning
BERT-Large
Natural Language Processing
PyTorch
Fine-Tuning
Yes (run faster on POD16)
BERT-Large Fine Tuning on IPU
BERT-Large
Natural Language Processing
Hugging Face
Fine-Tuning
Yes (run faster on POD16)
General-purpose Text Embeddings on the IPU - Inference
Sentence-T5, E5, ST MPNet
Natural Language Processing
Hugging Face
Inference
Yes (run faster on POD16)
SQuAD and MNLI on IPUs using DeBERTa - Inference
DeBERTa
Natural Language Processing
Hugging Face
Inference
Text Classification for sentiment analysis on IPU using multiple NLP models - Inference
BERT, RoBERTa
Natural Language Processing
Hugging Face
Inference
-
Text Generation on IPU with GPT2 - Inference
GPT2-S/M/XL
Natural Language Processing
PyTorch
Inference
Support for larger GPT2 models
Named Entity Recognition on IPU using BERT - Inference
BERT
Natural Language Processing
Hugging Face
Inference
-
Named Entity Recognition on IPU using BERT- Fine-Tuning
BERT
Natural Language Processing
Hugging Face
Fine-Tuning
Yes (run faster on POD16)
Multiple choice on IPU using RoBERTa - Fine-Tuning
RoBERTa
Natural Language Processing
Hugging Face
Fine-Tuning
Yes (run faster on POD16)
Question-Answering on IPU using RoBERTa - Fine-Tuning
RoBERTa
Natural Language Processing
Hugging Face
Fine-Tuning
Yes (run faster on POD16)
Text Summarization on IPU using BART-L - Inference
BART-L
Natural Language Processing
Hugging Face
Inference
-
Summarization on IPU using T5 Small - Fine-Tuning
T5 Small
Natural Language Processing
Hugging Face
Fine-Tuning
Yes (run faster on POD16)
Text Classification task on IPU using RoBERTa - Fine-Tuning
RoBERTa
Natural Language Processing
Hugging Face
Fine-Tuning
Yes (run faster on POD16)
Token Classification task on IPU using BERT - Fine-Tuning
BERT
Natural Language Processing
Hugging Face
Fine-Tuning
Yes (run faster on POD16)
Translation on IPU using BART-Base - Fine-Tuning
BART-Base
Natural Language Processing
Hugging Face
Fine-Tuning
Yes (run faster on POD16)
Fraud Detection on IPU using RGCN GNN - Training
RGCN
Graph Neural Network
PyG (PyTorch Geometric)
Training
Yes (run faster on POD16)
Training dynamic graphs on IPU using Temporal Graph Networks (TGN)
Temporal Graph Networks (TGN)
Graph Neural Network
PyG (PyTorch Geometric)
Training
-
Molecular property prediction on IPU using SchNet - Training
SchNet
Graph Neural Network
PyG (PyTorch Geometric)
Training
Yes (run faster on POD16)
Node Classification on IPU using Cluster-GCN - Training
Cluster-GCN
Graph Neural Network
PyG (PyTorch Geometric)
Training
-
Node Classification on IPU using Cluster-GCN - Training
Cluster-GCN
Graph Neural Network
TensorFlow 2
Training
-
Molecular property prediction on IPU using GIN - Training
GIN
Graph Neural Network
PyG (PyTorch Geometric)
Training
-
Training Neural Bellman-Ford networks (NBFnet) for inductive knowledge graph link prediction on IPUs
Bellman-Ford networks (NBFnet)
Graph Neural Network
PyG (PyTorch Geometric)
Training
Yes (run faster on POD16)
Molecular property prediction on IPU using GPS++ (OGB-LSC) - Inference
GPS++
Graph Neural Network
TensorFlow2
Inference
-
Molecular property prediction on IPU using GPS++ (OGB-LSC) - Training
GPS++
Graph Neural Network
TensorFlow2
Training
Yes (POD16 enables Training GPS++)
Link prediction training for knowledge graphs using Distributed KGE on IPU (OGB-LSC)
Dist KGE - TransE (256)
Graph Neural Network
PyTorch
Training
Yes (run faster on POD16)
Multitask Molecular Modelling with Graphium on the IPU
GCN, GIN, GINE
Graph Neural Network
PyG (PyTorch Geometric)
Training & Inference
-
Molecular Prediction on IPUs using MolFeat
MolFeat
Graph Neural Network
PyG (PyTorch Geometric)
Fine Tuning & Inference
-
Molecular Classification on IPUs using MolFeat - Fine-Tuning
MolFeat
Transformer
Hugging Face
Fine-Tuning
-
Automatic Speech Recognition (ASR) on IPU using wav2vec - Fine-Tuning
wav2vec
Speech Processing
Hugging Face
Fine-Tuning
Yes (run faster on POD16)
Automatic Speech Recognition (ASR) on IPU using wav2vec - Inference
wav2vec
Speech Processing
Hugging Face
Inference
Yes (run faster on POD16)
Multi-Horizon Financial Forecasting on IPU using DeepLOB-Seq2Seq - Training
DeepLOB
Time-Series
TensorFlow2
Training & Inference
-
Multi-Horizon Financial Forecasting on IPU using DeepLOB-Attention - Training
DeepLOB
Time-Series
TensorFlow2
Training & Inference
-
Tessellate IPU - Basics of Tile Mapping on Tensors and Vertices
Tutorial
-
-
-
-