Publications
Network World Models as Environments for Algorithm Design on Complex Systems
Preprint, 2026
An action-conditioned Network World Model that predicts how interventions spread on a network, used as a fast evaluator for a coding agent that designs algorithms for complex-system tasks.
CONSTRAINER: Promptable Graph-Structured Optimization via Constraint Conditioning
NeurIPS 2026
Promptable graph-structured optimization via constraint conditioning.
EpiWorld: Grounding LLM Policy Agents in Epidemiological World Models
EMNLP 2026 Findings
Grounding LLM policy agents in epidemiological world models.
Towards Systematic Generalization for Power Grid Optimization Problems
KDD 2026 (Oral)
Systematic generalization for power grid optimization problems.
Toward World Models for Epidemiology
ICLR 2026 Workshop on World Models: Understanding, Modelling and Scaling
Toward building world models for epidemiological modeling and simulation.
LUMINA: Foundation Models for Topology Transferable ACOPF
ICLR Workshop on Foundation Models for Science, 2026
Foundation model for topology-transferable AC optimal power flow.
On the Fundamental Limits of LLMs at Scale
Transactions on Machine Learning Research (TMLR), 2025
Accepted in Transactions on Machine Learning Research (TMLR).
Deep Causal Generative Models with Property Control
Preprint, 2025
Preprint on causal generative modeling with controllable properties.
LLM-Informed Discrete Prompt Optimization
ICML 2024 Workshop on LLMs and Cognition
Workshop paper on discrete prompt optimization informed by LLMs.
Content-aware Urdu handwriting generation
International Conference on Document Analysis and Recognition, 2023
Generating Urdu handwriting with content-aware methods.