← Search

Rajat Vadiraj Dwaraknath

4 accepted papers

2026

FlashSketch: Sketch-Kernel Co-Design for Fast Sparse Sketching on GPUs

ICML 2026oral

Sparse sketches such as the sparse Johnson–Lindenstrauss transform are a core primitive in randomized numerical linear algebra because they leverage random sparsity to reduce the arithmetic cost of sketching, while still offering strong approximation guarantees. Their random sparsity, however, is at…

Cited by 0SourceScholar
2025

SD-KDE: Score-Debiased Kernel Density Estimation

NeurIPS 2025poster

We propose a method for density estimation that leverages an estimated score function to debias kernel density estimation (SD-KDE). In our approach, each data point is adjusted by taking a single step along the score function with a specific choice of step size, followed by standard KDE with a modif…

Cited by 0SourceScholar
2023

Fixing the NTK: From Neural Network Linearizations to Exact Convex Programs

NeurIPS 2023poster

Recently, theoretical analyses of deep neural networks have broadly focused on two directions: 1) Providing insight into neural network training by SGD in the limit of infinite hidden-layer width and infinitesimally small learning rate (also known as gradient flow) via the Neural Tangent Kernel (NTK…

Cited by 0SourcePDFScholar