Profiling JAX/XLA with XProf in TensorBoard

Experimenting with Different JAX Precisions

JAX Device Mesh: Parallel Vision Transformer Code

Self-Supervised Learning Collapse: a Quick CIFAR-10 Experiment

I-JEPA Explained with a Single Batch Run

Self-Supervised Learning Review: From SimCLR to DINOv2

Inside DINOv2: Architecture Analysis + CIFAR-10 Experiment

Graph Contrastive Learning: Building MovieLens-100k Recommendations

Exploring LightGCN: A Movie Recommendations Experiment

MLP-Mixer Introduction + A CIFAR-10 Run

Hands-On with TAPIR: Point Tracking Experiment & Code Walkthrough

Fine-tuning Vision Transformers on TPU (ImageNet → CIFAR-10)

VGG-SfM and Mip-NeRF 360 Pipeline for iPhone Video 3D Reconstruction

Structure from Motion (SfM): From COLMAP to VGGSfM

Mip-NeRF 360 on TPUs: Setup, Training, and First Render!

Nerfstudio on Lightning AI GPUs

Radial Basis Function (RBF): The Most Popular Kernel in Machine Learning

How NeRF Works: Exploring a Tiny NeRF Code

Free TPU Access & JAX/PyTorch Setup with TPU Research Cloud

3D Gaussian Splatting: Optimization Explained & Viewer Demo

A Deep Dive into Swin Transformer Attention Maps

Analyzing Swin Transformer: A Code Walkthrough

Building a Cluster-GCN Model with JAX: A Step-by-Step Guide

Implementing GNN Neighbor Sampler in JAX: A Practical Guide

Graph Sampling for GNNs: A Tutorial

Graph Attention Networks with DGL

Understanding Mini-Batch Training in PyTorch Geometric

JAX JIT Compilation Explained: From Python to JAXPR

CNNs with Flax: A JAX Tutorial for Image Classification

Vision Transformers (ViT): A JAX Tutorial for Image Classification

Graph Attention Networks with JAX

JAX Conv Layer Explained

JAX and Flax: A Simple Neural Network

Training GCNs with PyG and Jraph: A Side-by-Side Comparison

Exploring Sparse Subspace Clustering: Theory and Practice

Graph Attention Networks with PyTorch Geometric

PyTorch Conv2d Explained

A Step-by-Step Guide to Spectral Clustering

ViT vs CNN: A Comparative Experiment

PyTorch code for GCN and SGC

Graph Convolutional Networks (GCNs) in PyTorch

Vision Transformers (ViT) pytorch code