|
PhD student in Electrical Engineering at Stanford, co-advised by Dr. Emily Fox and Dr. John M. Cioffi. I work on reinforcement learning for verifiable rewards, preference optimization, test-time scaling, and agentic systems for reliable reasoning models. Collaborations across Google DeepMind, Meta MSL, Amazon AGI, and Microsoft Core AI. Building alexein.ai, an agentic platform for collaborative AI agents. |
|
|
Preference Optimization & Test-Time Scaling Sample-efficient preference-learning and RL methods for aligning language-model reasoning, plus adaptive test-time compute that controls inference depth, tool invocation, and verification under strict budgets. I study how post-training and inference-time search jointly shape reasoning, and the trade-offs among accuracy, latency, and reliability.
|
|
Agentic System Architectures & Coding Agents Agentic systems that plan, act, and self-verify over long horizons, spanning multi-agent coordination and the software agents I develop at alexein.ai. At Amazon AGI I study how coding agents convert test-time compute into solved tasks by exploring diverse solutions and writing trustworthy verifiers, and how those behaviors are internalized through supervised and reinforcement post-training.
|
|
AI Alignment & Benchmarking Persona-based benchmarks and user-simulation frameworks that evaluate how models behave across diverse populations and interactive settings. I care about calibrated uncertainty, reducing sycophancy, and measurement methodology that makes alignment progress observable and reproducible.
|
| Venue | Title |
|---|---|
NeurIPS '26 |
General Preference Reinforcement Learning |
EMNLP '26 |
Continuous-Utility Direct Preference Optimization |
CoLM '26 |
Active Alignment with Bayesian General Preference Models |
AAAI '27 |
Sycophancy Disentanglement in LLMs via Reward Decomposition |
ICLR '27 |
Teaching Agents to Code Reliably · Amazon AGI |
ICLR '27 |
LoRe-KV: Lookahead Output-Perturbation for KV Cache Eviction |
NeurIPS '26 |
Epistemic Uncertainty for Test-Time Discovery |
CoLM '26 |
S³: Stratified Scaling Search for Test-Time in Diffusion LMs |
Preprint |
Coverage Preservation in RLVR as an Off-Policy Constraint |
Nature |
MatrAIx: Simulating the World with 8.3 Billion Persona Agents · invited |
JMLR '26 |
Graph Neural Networks for Accelerating Low-Rank SDP Solvers |
KDD '26 |
Neural Gaussian Radio Fields for Channel Estimation |
Full list, figures, and blog posts on ahmd-mohsin.github.io.
|
Fellowships & Medals
|
Service
|
Reinforcement Learning · LLM Post-Training · Preference Optimization · Test-Time Scaling · Agentic Systems · GRPO / LoRA / ZeRO


