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 and interested in Geometric Learning and Self Supervised Learning. Currently, I am working on:
I write about Geometric Deep Learning, Self-Supervised Learning and JAX.
Feel free to reach out via email at: sauravvmaheshkar [at] gmail [dot] com
Feel free to reach out via email at: sauravvmaheshkar [at] gmail [dot] com
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).