zhao-han/examples

Example code and applications for machine learning on Graphcore IPUs

Graphcore code examples

This repository contains sample applications and code examples for use with Graphcore IPUs.

If you are interested in finding out more about Graphcore, including getting preview access to IPUs to run these examples, please register your interest here.

Please note we are not currently accepting pull requests or issues on this repository. If you are actively using this repository and want to report any issues, please raise a ticket through the Graphcore support portal: https://www.graphcore.ai/support.

Repository contents

Notable examples

Example	Link
BERT	code
ResNext	code
EfficientNet	code
Recommendation using Autoencoders	code
Sales forecasting example	code
MCMC methods example	code
Recurrent layer kernel benchmarks	code
ResNet	inference code, training code
Constrastive Divergence VAE using MCMC methods	code
Grouped convolution kernel benchmarks	code
Example reinforcement learning policy model	code

Application examples

The applications/ folder contains example applications written in different frameworks targeting the IPU. See the READMEs in each folder for details on how to use these applications.

Code examples

The code_examples/ folder contains small code examples showing you how to use various software features when developing for IPUs. See the READMEs in each folder for details.

Utilities

The utils/ folder contains utilities libraries and scripts that are used across the other code examples. Currently this is split into:

• utils/tests - Common python helper functions for the repo's unit tests.
• utils/benchmarks - Common python helper functions for running benchmarks on the IPU in different frameworks.