Hello, I'm Nithin Balamurugan

Computer Science Student at University of Western Ontario | Graduating April 2027

AI Research Platform Developer | Co-author of "Tiny Recursive Models on ARC-AGI-1" Research Paper

Nithin Balamurugan

About Me

I'm a Computer Science student at Western University (April 2027) currently working in data engineering with Azure and Databricks.

I also did research on Tiny Recursive Models and co-authored an arXiv paper on ARC-AGI-1. Right now, I'm focused on breaking into Data Engineering and ML infrastructure full-time.

Data & Cloud (Core Focus)

Azure Azure Databricks Azure Data Lake Storage Azure Data Factory Azure SQL Azure DevOps Salesforce Datadog

Backend & Product Engineering

FastAPI Node.js Next.js React Docker OpenAI API Git/GitHub

Languages

Python SQL TypeScript JavaScript Java

Data/ML Libraries

Pandas NumPy PyTorch Hugging Face Vector Search Embeddings
Nithin candid photo 1
Nithin candid photo 2
Nithin candid photo 3

Work Experience

TD

Data Engineer

September 2026 - December 2026
  • Engineered and maintained Azure ETL pipelines processing 2,500-3,000 documents per ingestion cycle across Databricks, Azure SQL, and Data Lake.
  • Optimized AI data pipelines and embedding infrastructure, improving retrieval accuracy from 95% to 98% while reducing provisioned compute by 47%.
  • Built cloud infrastructure, CI/CD, and Datadog monitoring across DEV/TEST/PROD, enabling more reliable deployments and faster incident triage.

Key Stack: Azure, Azure Databricks, Azure SQL, Azure Data Lake Storage, CI/CD, Datadog

CIBC

Data Engineer

May 2026 - August 2026
  • Resolved 100+ production incidents via root-cause analysis, maintaining 98%+ uptime across 4+ systems spanning Azure, Databricks, and SQL.
  • Managed and improved Databricks queries and Azure Data Factory pipeline reliability to strengthen workflow stability in production.
  • Supported deployments and release management through Azure DevOps and Salesforce, improving production workflows and deployment reliability.

Key Stack: Azure, Azure Databricks, Azure Data Factory, Azure DevOps, Salesforce, SQL

Varonova Tech Inc.

AI Software Developer Intern

January 2026 - April 2026
  • Refactored an AI video generation pipeline into modular object-oriented Python architecture, separating scene proxy operations, depth processing, and rendering for maintainability and consistent workflows.
  • Implemented a 3D point cloud reconstruction pipeline using NumPy, with depth backprojection, 4x4 camera pose transformations, and voxel-based downsampling to generate optimized scene proxies.
  • Developed a novel-view rendering pipeline with z-buffer occlusion, point splatting, OpenCV inpainting, and FFmpeg H.264 export for downstream AI video generation workflows.

Key Stack: Python, NumPy, OpenCV, FFmpeg, 3D Rendering, Point Clouds

Western Cyber Society (club)

Software Developer

September 2025 - May 2026
  • Designed and developed LedIq, a functional replica of a top-5 banking system on IBM z/OS using COBOL, JCL, SQL, and REXX, integrated with a Node.js frontend and FastAPI backend for low-latency risk evaluation.
  • Deployed an ONNX-hosted XGBoost model for real-time risk scoring and added SHAP explainability, persisting predictions and feature-level explanations in IBM Db2 for auditability.

Key Stack: IBM z/OS, COBOL, JCL, Node.js, FastAPI, XGBoost, SHAP, Db2

CognitomeAI

Software Engineer

September 2025 - December 2025
  • Built a Python/FastAPI pipeline to ingest, clean, index, and retrieve PhD and peer-reviewed papers using NLP, embeddings, vector search, and semantic similarity.
  • Developed an interactive data visualization that clusters and ranks papers by relevance using statistical analysis and similarity scoring to support faster data-driven decisions.
  • Built and deployed a GPT-3.5 RAG chatbot using Python, Docker, AWS, and Next.js for citation-backed research workflows.

Key Stack: Python, FastAPI, NLP, Embeddings, Vector Search, GPT-3.5, Docker, AWS, Next.js

Research

Tiny Recursive Models on ARC-AGI-1

Inductive Biases, Identity Conditioning, and Test-Time Compute

  • Built reproducible data pipelines for ARC-AGI-1 experiments, processing and evaluating 400+ tasks at scale.
  • Engineered a 1,000-sample data augmentation pipeline, improving Pass@1 accuracy by 10.75 percentage points.
  • Ran data analysis and model ablation experiments, identifying puzzle-ID embeddings as a critical performance dependency.
  • Showed strict puzzle-ID dependence: accuracy dropped from 40.00% to 0.00% when IDs were blanked or randomized under the verification protocol.
  • Found shallow effective recursion: step-1 already reached 38.25% Pass@1 (94.4% of final), with accuracy saturating by step 4 and unchanged at step 6.
  • Benchmarked efficiency on H100: TRM (7M) used 2.4 GB VRAM at 31.3 samples/s vs Llama 3 8B QLoRA at 6.1 GB and 0.24 samples/s.

Impact: This paper currently has 3 citations.

Featured Projects

StoryOS Video Orchestration Platform

StoryOS – Deterministic Video Orchestration

Working on a demo project for a media company to solve inconsistency issues in AI video regeneration. My role focuses on 3D rendering and point cloud technology to ensure consistent character and scene representation across video generations, eliminating the randomness that plagues current AI video tools.

3D Rendering Point Clouds Video Generation Python AI/ML
AI Finance Tracker Dashboard

AI Finance Tracker

Built a full-stack AI finance application using React, Node.js, and PostgreSQL to track expenses, budgets, and spending, with secure user authentication and account management via Supabase, including protected routes, session handling, and user-specific data isolation. Developed a Python-based AI backend (FastAPI) that leverages Pandas for data analysis and OpenAI GPT-3.5-turbo for natural language processing, implementing a hybrid AI system for personalized financial insights.

React Node.js PostgreSQL Python FastAPI OpenAI Supabase Docker
CSV Converter Project

CSV Converter

Developed a high-accuracy PDF-to-CSV conversion algorithm using Python that extracts every table into separate CSV files, generates a full-document text file, and outputs structural metadata files detailing table outlines, dimensions, and formatting, robust across any document layout. Built a modular extraction pipeline using Docling, PyMuPDF, Pandas, and a 3-layer OCR stack.

Python Pandas PyMuPDF OCR
Tiny Recursive Models Research Paper

Tiny Recursive Models Research Paper

Co-authored "Tiny Recursive Models on ARC-AGI-1: Inductive Biases, Identity Conditioning, and Test-Time Compute", analyzing the behavior of Tiny Recursive Models (TRMs) on the ARC-AGI-1 benchmark. Performed empirical ablations and efficiency analyses to isolate the impact of test-time compute, puzzle-identity conditioning, and recursion depth on model performance. Benchmarked TRMs against a QLoRA-fine-tuned LLaMA 3 8B baseline. (PAPER LINK AT THE END OF THE PAGE)

Research AI/ML Python PyTorch

Personal Portfolio Website

Developed a personal portfolio website to showcase projects in depth and share background and skills. Implemented using Next.js, React, TypeScript, server-side rendering (SSR), React Hooks, Tailwind CSS, PostCSS, responsive design, dark mode, animations, form handling, ESLint, Node.js, npm, and Next.js Google Fonts.

Next.js React TypeScript Tailwind CSS PostCSS Node.js

BookBuds

Built a full-stack mobile reading log application using React Native and Expo, enabling chapter-level note-taking with offline-first persistence via AsyncStorage and cross-platform support for iOS and Android. Implemented social features using Firebase Authentication and Cloud Firestore, including user accounts, friend connections, and real-time commenting on shared chapter notes. Integrated an AI chatbot to summarize and clarify user notes.

React Native Expo Firebase Firestore AsyncStorage AI
OnlyUs

OnlyUs

Built a cross-platform app focused on bridging long-distance relationships using Expo, React Native, and TypeScript, with Expo Router handling navigation across iOS, Android, and web. Designed a Supabase backend with Postgres, auth, storage, and real-time sync to power posts, letters, and calendar journaling features.

React Native Expo TypeScript Supabase Expo Router Nativewind Reanimated AsyncStorage
GraphMind

GraphMind

Originally built as a VM-hosted Docker stack (FastAPI, Celery, RabbitMQ, Redis, Neo4j) with a Next.js frontend to convert unstructured content into queryable knowledge graphs. To avoid entering payment details for cloud VM providers, I rebuilt it as a no-card demo architecture on Vercel + AuraDB + Upstash while keeping the same core extraction flow.

Next.js TypeScript Node.js FastAPI Celery RabbitMQ Redis OpenAI Knowledge Graph AuraDB Upstash
FixMyResume

FixMyResume

Full-stack AI LaTeX resume builder that converts resume PDFs to editable structured content and exports clean LaTeX for Overleaf. Users can upload a PDF, edit parsed entries, enhance bullets with AI, and generate ATS-friendly output quickly.

FastAPI Python Groq API Docling PyMuPDF LaTeX

LedIq

Functional replica of a top-5 banking system on IBM z/OS mainframe. Implemented core components with COBOL, JCL, SQL, and REXX. Integrated Node.js frontend with FastAPI backend for real-time credit risk evaluation using XGBoost and explainable AI (SHAP).

COBOL IBM z/OS Node.js FastAPI XGBoost SHAP
Self Watering Flower Pot Project

Self Watering Flower Pot

Built an automated self-watering plant system using Arduino to monitor soil moisture and trigger watering below a set threshold, enabling plant care during long absences. Integrated sensors and output modules (LCD, moisture sensor, LED, buzzer, water pump) using Java and Matlab to display moisture data and signal watering events.

Java MATLAB Arduino IoT Hardware

Get In Touch

I'm always open to discussing new projects, creative ideas, or opportunities to be part of your vision. Feel free to reach out!