Graduate Student · Stanford · SAIL
Co-advised by Dr. Emily Fox and Dr. John M. Cioffi
Research on preference optimization and alignment for LLMs, adaptive test-time scaling and discovery, and self-evolving multi-agent systems.
Graduate student at Stanford — LLM post-training, test-time scaling, and reinforcement learning. Collaborations with Google DeepMind, Meta, Amazon AGI, and Microsoft Core AI.
My research spans preference optimization and alignment for LLMs, adaptive test-time scaling and discovery, and interactive multi-agent systems — with collaborations across Google DeepMind, Meta, Amazon AGI, and Microsoft Core AI.
Preference Optimization and Alignment for LLMs
ICML'26 · CoLM'26 · NeurIPS'26 · EMNLP'26Designing sample-efficient preference learning and reinforcement-learning methods for aligning language-model reasoning: continuous-utility and general-preference formulations (CU-DPO, General Preference RL), active alignment under Bayesian general preference models with calibrated uncertainty, and sycophancy reduction.
Test-Time Scaling and Test-Time Discovery
ICML'26 · CoLM'26 · NeurIPS'26 · EMNLP'26Developing adaptive test-time compute methods for reasoning and scientific discovery under uncertainty: dynamic control of inference depth, tool invocation, and verification under strict budgets; stratified scaling search for diffusion language models; and epistemic-uncertainty-driven test-time training for discovery.
Interactive and Multi-Agent Systems
NeurIPS'26 · OngoingBuilding self-evolving multi-agent ecosystems in which agents accumulate scoped memory, earn reputation through Bayesian posteriors, and coordinate over an evolving social graph. Related work on learning from code-agent trajectories via causal redundancy analysis, and on privacy, security, and shared context in collaborative agentic reasoning.
Stanford Artificial Intelligence Laboratory (SAIL)
Sep 2025 – Present
Prof. Emily Fox
Internet of Evolving Agents
Test-Time Compute and Reasoning in Large Language Models
Bayesian Preference Alignment for Mathematical Reasoning
Intel Corporation
Sep 2024 – Dec 2024
Dr. John M. Cioffi
Neural Gaussian Radio Fields for Environment Perception
Samsung Semiconductors
Jun 2024 – Sep 2025
Dr. John M. Cioffi
Graph Neural Networks for Accelerating Low-Rank SDP Solvers
Muhammad Ahmed Mohsin, Muhammad Umer, Ahsan Bilal, John M. Cioffi, Emily Fox
Z. Ali*, M. A. Mohsin*, M. Umer, A. Bilal, E. Fox

IEEE ICASSP 2026
Barcelona, Spain
Presenting work on diffusion models and non-stationary channel estimation
Knight-Hennessy Fellowship finalist.
Served as an Area Chair for ICASSP and NeurIPS 2025.
Selected as a Qualcomm Fellowship finalist.
Serving as Workshop Co-Chair for VTC Fall 2026 in Boston.
Served as a member of the Technical Program Committee at NeurIPS 2025 and also as a NeurIPS reviewer.
Received the Exemplary Reviewer recognition for IEEE Wireless Communications Letters 2025.
Added as a founding member of the IEEE Special Interest Group on AI-driven TN-NTN Networks.
Paper accepted at ICML 2025 on "Continual Learning for Wireless Channel Estimation," along with a student travel grant to ICML.
Received an ICC student travel grant for Montreal and a best workshop paper award for RAG-optimized wireless environment perception.
Two papers on Hierarchical Deep RL and Joint Source Compression accepted at AAAI 2025 in Philadelphia.
Two papers on diffusion-based Langevin dynamics and minPMAC optimization accepted at IEEE ICASSP 2025 in India.
Awarded a Globecom 2024 travel grant for travel to Cape Town.
Received a Best Poster Award nomination at the 6G Summit in Abu Dhabi.
Awarded the Rector's Gold Medal for best undergraduate thesis.
Accepted to Stanford with the Stanford Graduate Fellowship.
First paper accepted at ICDAR 2023, outperforming Microsoft's DiT on table recognition tasks.
Received the ECAT Scholarship for ranking among the top 10 in Pakistan in the engineering category test.
Received the President's Medal for ranking among the top 3 students across Pakistan at the HSSC level.
Awarded the NUST scholarship for maintaining a 4.0 GPA.
Conference Reviewer
Journal Reviewer
Technical Program Committee
Leadership
Awards & Recognitions
—Stanford Graduate Fellowship
—Knight-Hennessy Fellowship — Finalist
—President's Medal — Third position nationwide in pre-engineering
—6G Summit Abu Dhabi — Best Poster Nomination
—IEEE ICC Canada — Best Workshop Paper Award
—IEEE Communications Society Competition — Honorary Mention
—IEEE FIT 2026 — Best Main Conference Paper Award
—Rector's Gold Medal — Best final year project cohort (NUST, 2024)
Travel Grants
I am always open to discussing new research collaborations and opportunities.