OpenPi-GPU SHARDER
Jun 2026, Research
An implementation of a multi-node FSDP for the OpenPI repo.
Arjun Virk
- Software Engineering @ UWaterloo
- Robotics Research Engineer @ BracketBot
- Computational Oncology Researcher @ UCLA BAIR
- Prev. Data and Transformer Researcher @ Triage Technologies ($50M)
- Silver Medalist at Canada’s National Science Fair
- ML Agents and Research Consulting Business
- Worked with Google, Microsoft, Verizon, Interac for ML/Datacentre Consulting
Work
Short Essays, Literature Reviews, Books.
Machine Learning Bible
May 2026, Writing
A 200-page handwritten guide breaking down the math, intuition, and architectures behind modern machine learning and deep learning systems.
KhemKernel: Transformer from Scratch in numpy + Custom CUDA Kernels
Apr 2026, Research
Built an encoder-decoder Transformer from scratch in NumPy and C++, using 5,000+ handwritten lines and custom CUDA kernels for FlashAttention to translate SMILES into IUPAC names.
Robotics Bible
Jul 2026, Writing
A full guide of ML in Robotics.
LoRA VLA Research @ BracketBot
May 2026, Research
Developing novel LoRA and DoRA adapter architectures for efficient VLA fine-tuning, robotic manipulation, and continual learning under Isaac Bautista at BracketBot. (more soon)
ECE 657: Course Collaborator
Jun 2026, Teaching
Collaborating with Prof. Amir-Hossein Karimi on Machine Learning Fundamentals at UWaterloo Graduate ECE Department. More soon.
Failure-Aware ACT (FAACT)
Mar 2026, Research
Built a failure-aware action chunking Transformer that detects trajectory failures with minimal pretraining and autonomously recovers through a self-correction algorithm.
Sweep JEPA
Jun 2025, Research
Worked with Dr. William Speier and Ashwath Radhachandran, PhD, on JEPA and ViT-based models to improve thyroid cancer risk stratification beyond ACR TI-RADS. (more soon)
SOTA Pan-Cancer Immunotherapy Quantification
Apr 2024, Research
Developed RL PPO system to predict immunotherapy efficacy and recommend ICI inhibitors across cancer types from 15% to 80%. Silver Medal at Canada’s National Science Fair.
Inference Engineering: A Deep Dive
Jun 2026, Writing
A deep dive into LLM inference and GPU architecture.
Explanation Tuning Research (Triage)
Jun 2024, Engineering
Worked under Mat Kallada from the Mila Institute to organize medical sequence data pipelines and research explanation-guided tuning for medical vision-language transformers. (more soon)
BigBrother: First Autonomous Interactive Agentic Browser
Jan 2026, Engineering
Built an autonomous agentic browser extension that completes web tasks from natural-language prompts. Awarded $200 in prizes.
Wisp: Custom MCP for Daily Notion and Google Workspace
Nov 2025, Engineering
Led the implementation of a custom MCP server that exposes Notion and Google Workspace as structured, machine-readable tools instead of unstructured text context.
Meridian
Jul 2026, Engineering
An app that logs what you are doing every ten minutes, learns your peak productivity hours, and builds a dynamic daily schedule around them.
Dancing Humanoid (1st Place Provincially)
Jul 2022, Engineering
Designed the CAD, wiring schematics, and C++ control system for a music-synced humanoid robot that performed a choreographed dance routine. Awarded $200 in prizes.
Backtracking
Jul 2026, Engineering
An app that fights the forgetting curve by tracking what you have learned and tuning the questions it asks each day so the material sticks.
Hack49: 700K Reached, $13K Funding
Jun 2024, Leadership
Led a 13-person team to reach 700K+ students, secure $13K in funding, recruit judges from Amazon, Elegoo, and Dell, and scale Hack49 into a global student hackathon.
Daybreak
Jul 2026, Engineering
A personal daily digest of paper recommendations, relevant news, and takes from the leaders you follow on X, built as a calmer replacement for endless scrolling.
Robot See, Robot Do
Feb 2026, Writing
Wrote a viral technical article explaining the math behind imitation learning, Action Chunking Transformers, and DAgger for robotics learning.