graphcore/poptorch

PyTorch interface for the IPU

/ 100

Emerging

This project helps machine learning engineers accelerate their PyTorch models on Graphcore's IPU hardware. It takes your existing PyTorch or PyTorch Geometric model code and, after minimal changes, compiles and runs it efficiently on the IPU, dramatically speeding up training and evaluation. It's designed for deep learning practitioners and researchers who are building and deploying large-scale neural networks.

180 stars. No commits in the last 6 months.

Use this if you are a machine learning engineer working with PyTorch or PyTorch Geometric and want to leverage Graphcore IPUs for faster model development and deployment.

Not ideal if you are not using Graphcore IPUs, or if you prefer a different deep learning framework like TensorFlow or JAX.

deep-learning-acceleration neural-network-training graph-neural-networks AI-model-deployment high-performance-computing

Stale 6m No Package No Dependents

Maintenance 0 / 25

Adoption 10 / 25

Maturity 16 / 25

Community 12 / 25

How are scores calculated?

Stars

180

Forks

Language

C++

License

MIT

Higher-rated alternatives

pymc-devs/pytensor

PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions...

arogozhnikov/einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

lava-nc/lava-dl

Deep Learning library for Lava

tensorly/tensorly

TensorLy: Tensor Learning in Python.

tensorpack/tensorpack

A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility

Explore ML Frameworks

All categories Trending ML Framework directory Insights