/notebooks

Small, standalone, machine learning & IPU notebooks

Primary LanguageJupyter NotebookMIT LicenseMIT

Research Notebooks

A collection of notebooks developed for researchers, designed for use with IPUs.

Path Description
sparsity_benchmarks Sparse matrix multiplication benchmarks (paper on arxiv)
moe_charlm Text generation using Mixture of Experts
hello_pytorch Minimal PyTorch example
tiny_stories Pre-training on TinyStories
compressed_dlrm Embedding compression for Recommendation models

For testing locally and contributing, see development.md.

License

Copyright (c) 2023 Graphcore Ltd. Licensed under the MIT License.

The included code is released under an MIT license, (see LICENSE).