Vivek Bharadwaj

I am an EECS graduate student at UC Berkeley, advised by the fantastic pair Jim Demmel and Aydın Buluç. I'm associated with the BeBOP and PASSION groups, and I'm supported by a Department of Energy Computational Science Graduate Fellowship from the Krell Institute.

Interests

Geometric Deep Learning
Sparse Computations
Randomized Sketches
Graph Algorithms

Background

PhD in Computer Science.
UC Berkeley, 2020-Present
BS in CS + Mathematics.
Caltech, 2016-2020

Email: firstname_lastname [at] berkeley.edu

Interests

Geometric Deep Learning
Sparse Computations
Randomized Sketches
Graph Algorithms

Background

PhD in Computer Science.
UC Berkeley, 2020-Present
BS in CS + Mathematics.
Caltech, 2016-2020

Email: firstname_lastname [at] berkeley.edu

news

Jul 12, 2025 07/12/25	On July 21 at 10AM PDT, I will give my 🎉 🎉 dissertation exit talk 🎉 🎉! You are welcome to attend, either virtually or in-person on UC Berkeley's campus. Find the details here.
Jun 22, 2025 06/22/25	I'm presenting this poster at LBNL's Deep Learning for Science workshop - stop by to chat if you're attending.
Apr 5, 2025 04/05/25	OpenEquivariance was accepted to SIAM ACDA 2025! See you in Montréal.

recent & upcoming talks

Jul 15, 2025

DOE CSGF Program Review, Washington, D.C.

Final Year Fellow Presentation (Chemistry Graph Neural Networks)

Slides

Mar 4, 2025

SIAM CSE 2025 Matrix Computations Minisymposium, Fort Worth, TX

Engineering Fast Kernels for Rotation-Equivariant Deep Neural Networks

Slides

Nov 1, 2024

Redwood Center for Theoretical Neuroscience Group Meeting, Berkeley, CA

Subspace Embeddings for Khatri-Rao Products (Redux)

Slides Video

selected publications

* denotes equal contribution. See more papers on the dedicated publications page.

UCB Thesis

Accelerating Multilinear Maps and Structured Sparse Tensor Kernels

Vivek Bharadwaj

University of California, Berkeley, 2025

Bib Pdf Video Slides

@phdthesis{bharadwaj_dissertation,
    author = "Bharadwaj, Vivek",
    title = "Accelerating Multilinear Maps and Structured Sparse Tensor Kernels",
    school = "University of California, Berkeley",
    year = "2025",
    url = "https://www2.eecs.berkeley.edu/Pubs/TechRpts/2025/EECS-2025-146.pdf"
}

ACDA

An Efficient Sparse Kernel Generator for O(3)-Equivariant Deep Networks

Vivek Bharadwaj*, Austin Glover*, Aydın Buluç, James Demmel

In Proceedings of the Conference on Applied and Computational Discrete Algorithms (ACDA), July 2025

Bib Pdf Code Poster

@inproceedings{openequivariance,
    author = "Bharadwaj, Vivek and Glover, Austin and Buluç, Aydın and Demmel, James",
    title = "An Efficient Sparse Kernel Generator for O(3)-Equivariant Deep Networks",
    booktitle = "Proceedings of the Conference on Applied and Computational Discrete Algorithms (ACDA)",
    chapter = "",
    pages = "32-46",
    year = "2025",
    month = "July",
    doi = "10.1137/1.9781611978759.3",
    url = "https://epubs.siam.org/doi/abs/10.1137/1.9781611978759.3"
}

NeurIPS

Efficient Leverage Score Sampling for Tensor Train Decomposition

Vivek Bharadwaj*, Beheshteh Rakhshan*, Osman Malik, Guillaume Rabusseau

In Advances in Neural Information Processing Systems, Dec 2024

Bib Pdf Code

@inproceedings{bharadwaj2024efficientleveragescoresampling,
    author = "Bharadwaj, Vivek and Rakhshan, Beheshteh T. and Malik, Osman Asif and Rabusseau, Guillaume",
    title = "Efficient Leverage Score Sampling for Tensor Train Decomposition",
    booktitle = "Advances in Neural Information Processing Systems",
    month = "Dec",
    year = "2024",
    eprint = "2406.02749",
    archivePrefix = "arXiv",
    primaryClass = "cs.DS"
}

NeurIPS

Fast Exact Leverage Score Sampling from Khatri-Rao Products with Applications to Tensor Decomposition

Vivek Bharadwaj, Osman Malik, Riley Murray, Laura Grigori, Aydın Buluç, James Demmel

In Advances in Neural Information Processing Systems, Dec 2023

Bib Pdf Code Poster Slides Web

@inproceedings{bharadwaj23_fast_krp_leverage,
    author = "Bharadwaj, Vivek and Malik, Osman Asif and Murray, Riley and Grigori, Laura and Buluç, Aydın and Demmel, James",
    editor = "Oh, A. and Naumann, T. and Globerson, A. and Saenko, K. and Hardt, M. and Levine, S.",
    booktitle = "Advances in Neural Information Processing Systems",
    pages = "47874--47901",
    publisher = "Curran Associates, Inc.",
    title = "Fast Exact Leverage Score Sampling from Khatri-Rao Products with Applications to Tensor Decomposition",
    volume = "36",
    month = "Dec",
    year = "2023"
}

IPDPS

Distributed-Memory Sparse Kernels for Machine Learning

Vivek Bharadwaj, Aydın Buluç, James Demmel

In 2022 IEEE International Parallel and Distributed Processing Symposium, Jun 2022

Bib Pdf Video Code Poster Slides

@inproceedings{bharadwaj22_distspml,
    author = "Bharadwaj, Vivek and Buluç, Aydın and Demmel, James",
    booktitle = "2022 IEEE International Parallel and Distributed Processing Symposium",
    title = "Distributed-Memory Sparse Kernels for Machine Learning",
    year = "2022",
    volume = "",
    issn = "",
    pages = "47-58",
    keywords = "machine learning algorithms;costs;runtime;collaborative filtering;layout;machine learning;inference algorithms",
    doi = "10.1109/IPDPS53621.2022.00014",
    url = "https://doi.ieeecomputersociety.org/10.1109/IPDPS53621.2022.00014",
    publisher = "IEEE Computer Society",
    address = "Los Alamitos, CA, USA",
    month = "jun"
}