Hi, my name is

Ajay Jagannath.

I build systems that understand.

I am a Master's student at Texas A&M University bridging the gap between scalable systems engineering and interpretable AI. My work focuses on Large Language Models (LLMs), Chain-of-Thought reasoning, and multimodal representation learning.

01. Selected Research

Unified Augmented Learning for Cross-Modal Representations (AmCLR)

Proposed a memory-efficient framework (AmCLR/xAmCLR) for bimodal contrastive learning. Achieved superior retrieval performance on CC3M using significantly smaller batch sizes (128) compared to CLIP baselines.

PyTorch Contrastive Learning CUDA Optimizations CLIP

Computing Neuron-Level Importance in CoT Reasoning

Developed a mechanistic interpretability pipeline to identify reasoning-critical neurons in LLMs. Enhanced these neurons to achieve a 14% accuracy boost on the GSM8K benchmark.

Mechanistic Interpretability LLM reasoning Neuron Attribution Chain of Thought

02. Technical Projects

PEFT for Diffusion Style Transfer

Engineered a dual-model regularization framework to fine-tune SDXL using LoRA and SaRA. Improved style alignment by 19% (CLIP Score) while preventing catastrophic forgetting.

Stable Diffusion XL LoRA/SaRA Model Distillation

Demand-Paged Virtual Memory Manager

Implemented a minimal x86 kernel in C++ featuring a complete virtual memory subsystem, recursive page table mapping, and a preemptive multi-threaded scheduler.

Low-level Programming x86 Assembly OS Kernel Architecture

03. Experience

Software Engineer @ Standard Chartered Global Business Services July 2022 - July 2024
Intern @ Standard Chartered Global Business Services May 2021 - July 2021