Skip to content

Author

Raaz Dwivedi

Harvard University, MIT

We have 4 of 51 papers

We haven’t gathered this author’s papers yet. Follow them and we’ll fetch their work.

Not the right person? Other researchers publish under this name.

#machine learning Preprint Jun 2025

N$^2$: A Unified Python Package and Test Bench for Nearest Neighbor-Based Matrix Completion

This paper introduces N$^2$, a unified Python package and testbed that consolidates a broad class of NN-based methods through a modular, extensible interface and introduces a new NN variant that achieves state-of-the-art results in several settings.

Caleb Chin, Aashish Khubchandani, Harshvardhan Maskara et al. · 0 citations

ORCA-bench: How Ready Are Language Model Agents for Oncall?

The ORCA-bench benchmark is introduced, a benchmark that puts general-purpose coding agents in a production-fidelity oncall setting and is a lower bound on the engineering investment required before frontier coding agents can be safely entrusted with production reliability.

Albert Gong, Kyuseong Choi, Abhineet Agarwal et al. · 1 citation
#machine learning Open access Nov 2022

Doubly robust nearest neighbors in factor models

An improved variant of nearest neighbors (NN) for estimation with missing data in latent factor models that provides a (near-)quadratic improvement in the non-asymptotic error and admits a significantly narrower asymptotic confidence interval when compared to both unit-unit or time-time NN.

Raaz Dwivedi, Katherine Tian, Sabina Tomkins et al. · 11 citations
#machine learning Preprint Oct 2025

One Pipeline, Many Transformers: Pattern-Specific Imputation Specialists for Tabular Missing Data

A single pre-training pipeline that builds transformer-based imputation specialists through three components: an entry-wise featurization that recasts imputation as supervised prediction over row--column context, a synthetic data generator with pluggable missingness modules, and prior-data fitting on millions of synthe...

Jacob Feitelberg, Dwaipayan Saha, Kyuseong Choi et al. · 2 citations · ⚡1

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.