Writing a Training Loop in JAX + FLAX Permalink
A simple end-to-end training and evaluation pipeline in JAX, Flax and Optax.
Hi there, My name is Saurav.
I am a Computer Science student at the University of Manchester interested in Geometric Learning and Self Supervised Learning.
Recent Posts
An Introduction to Convolutional Graph Neural Networks Permalink
A beginner friendly introduction to Convolutional Graph Neural Networks (GCNs).
An Introduction to Graph Attention Networks Permalink
A beginner friendly introduction to Attention based Graphical Neural Networks (GATs).
How to Create a Image Classification Model in JAX/Flax Permalink
Learn how to create a simple image classification model in Flax in this short tutorial complete with code and interactive visualizations.
An Introduction to Message Passing Graph Neural Networks (GNNs) Permalink
A beginner friendly introduction to Message Passing Graph Neural Networks (MPGNNs).