Customer-obsessed science
Research areas
-
July 30, 20268 min readInstead of compromising among parameter updates dictated by different training objectives, ControlG allocates computational capacity to objectives sequentially and dynamically.
-
-
July 9, 202610 min read
-
Featured news
-
ACL 2026 Findings2026Current AI-powered code assistance tools often struggle with poorly-defined problem statements that lack sufficient task context and requirements specification. Recent analysis of software engineering agents reveals that failures on such underspecified requests are highly correlated with longer trajectories involving either over-exploration or repeated attempts at applying the same fix without proper evolution
-
2026Long-horizon motion forecasting for multiple autonomous robots is challenging due to nonlinear agent interactions, compounding prediction errors, and continuous-time evolution of dynamics. Learned dynamics of such a system can be useful in various applications such as travel time prediction, prediction-guided planning and generative simulation of warehouse robots. In this work, we aim to develop an efficient
-
ICML 2025 Workshop on Foundation Models for Structured Data2026Tabular foundation models like TabPFN and TabICL achieve state-of-the-art performance through in-context learning, yet their architectures remain fundamentally opaque. We introduce KernelICL, a framework to enhance tabular foundation models with quantifiable sample-based inspectability. Building on the insight that in-context learning is akin to kernel regression, we make this mechanism explicit by replacing
-
2026Large language model (LLM) agents increasingly operate in streaming case-based reasoning (CBR) settings, where continuous improvement from past experience is crucial. Existing methods achieve this by storing past cases and retrieving similar ones as few-shot examples. This strategy fails near decision boundaries, where highly similar cases have conflicting outcomes and the discriminative factors are not
-
ICML 2026 Workshop on Deep Learning for Code (DL4C)2026Recent agentic approaches to LLM-based kernel generation have achieved strong results on CUDA, yet emerging AI accelerators such as AWS Trainium and Inferentia remain unaddressed. Writing kernels for these chips via the Neuron Kernel Interface (NKI) is particularly challenging due to a multi-engine architecture, tile-based programming with a fixed 128-element partition dimension, and explicit memory management
Collaborations
View allWhether you're a faculty member or student, there are number of ways you can engage with Amazon.
View all