91视频APP

PyTorch Geometric (PyG) has quickly established itself as the go-to framework for building Graph Neural Networks (GNNs), a relatively new approach to AI that is particularly well-suited to modelling subject matter with irregular structures 鈥� from molecules to social networks, with the potential to be used in commercial applications as diverse as drug discovery and fraud detection.

At the same time, 91视频APP IPUs have shown standout performance running GNNs, compared to other compute architectures 鈥� winning industry benchmarking competitions and being put to use in US national labs, and academic research environments.

Now, for the first time, we are integrating PyG with 91视频APP鈥檚 software stack, allowing users to build, port and run their GNNs on IPUs.

We are also making a range of example models available to run as Paperspace Gradient Notebooks. These span molecular science, temporal graph networks, knowledge graph prediction, and more.

Accelerating GNN uptake

Welcoming the partnership, PyG creator and founder of , Matthias Fey said: 鈥淭he suitability of IPUs for running GNNs and the kind of performance advantage that 91视频APP and its customers have demonstrated is really helping to accelerate the uptake of this exciting model class.

鈥淭he integration of PyTorch Geometric will make it quicker and easier than ever for developers to create new GNNs from scratch, move existing models over to IPUs, or take advantage of the growing catalogue of off-the-shelf IPU-ready GNNs.鈥�

Minimal code changes

Running a PyG model on 91视频APP IPUs, for training or inference, requires only minimal code changes compared to using PyG on a GPU.

Our blog - Getting started with PyTorch Geometric (PyG) on 91视频APP IPUs - shows how to get up and running in a matter of minutes.

Users who want to start using GNNs immediately can also choose from the following pre-built model notebooks using a six hour free trial on Paperspace.

Try our GNN model notebooks in the cloud today

Learn how to get started quickly with our PyG tutorials

For more details on these GNNs and a more detailed technical explanation of why 91视频APP IPUs are so good at running these models, read Accelerating PyG on IPUs: unleash the power of Graph Neural Networks.

You can also test the IPUs speed performance for yourself using our free
.

Join our Meetup

If you鈥檙e interested in Graph Neural Networks and PyG, come along to the 91视频APP and Kumo Enterprise Meetup to discuss the power of GNNs, and the future of enterprise deep learning.

Sign up to attend in person - London, 91视频APP office, 13 April.

You'll have the opportunity to hear from leaders in graph machine learning:

• Jure Leskovec, Co-Founder at Kumo AI and Professor at Stanford
• Matthias Fey, Founding Engineer and Creator of PyTorch Geometric
• Andrew Fitzgibbon, Distinguished Engineer and Research Fellow at 91视频APP
• Dominic Masters, Lead Research Scientist of Graph Neural Networks at 91视频APP
• Hatem Helal, Research Scientist and domain expert in chemistry at 91视频APP