Rohan Yadav

All Publications

• Task-Based Tensor Computations on Modern GPUs PROCEEDINGS OF THE ACM ON PROGRAMMING LANGUAGES-PACMPL Yadav, R., Garland, M., Aiken, A., Bauer, M. 2025; 9 (PLDI)

  View details for DOI 10.1145/3729262

  View details for Web of Science ID 001532151900005

• Composing Distributed Computations Through Task and Kernel Fusion Yadav, R., Sundram, S., Lee, W., Garland, M., Bauer, M., Aiken, A., Kjolstad, F., ACM ASSOC COMPUTING MACHINERY. 2025: 182-197

  View details for DOI 10.1145/3669940.3707216

  View details for Web of Science ID 001481633300012

• Improving Parallel Program Performance with LLM Optimizers via Agent-System Interfaces Wei, A., Nie, A., Teixeira, T. X., Yadav, R., Lee, W., Wang, K., Aiken, A. edited by Singh, A., Fazel, M., Hsu, D., Lacoste-Julien, S., Berkenkamp, F., Maharaj, T., Wagstaff, K., Zhu, J. JMLR-JOURNAL MACHINE LEARNING RESEARCH. 2025: 66155-66177

  View details for Web of Science ID 001693167600236

• KDRSolvers: Scalable, Flexible, Task-Oriented Krylov Solvers Zhang, D., Yadav, R., Aiken, A., Kjolstad, F., Treichler, S., ACM ASSOC COMPUTING MACHINERY. 2025: 1351-1365

  View details for DOI 10.1145/3731599.3767501

  View details for Web of Science ID 001661298800147

• Automatic Tracing in Task-Based Runtime Systems Yadav, R., Bauer, M., Broman, D., Garland, M., Aiken, A., Kjolstad, F., ACM ASSOC COMPUTING MACHINERY. 2025: 84-99

  View details for DOI 10.1145/3669940.3707237

  View details for Web of Science ID 001481633300006

• DISTAL: The Distributed Tensor Algebra Compiler Yadav, R., Aiken, A., Kjolstad, F. edited by Jhala, R., Dillig ASSOC COMPUTING MACHINERY. 2022: 286-300

  View details for DOI 10.1145/3519939.3523437

  View details for Web of Science ID 000850435600020

• SpDISTAL: Compiling Distributed Sparse Tensor Computations Yadav, R., Aiken, A., Kjolstad, F., IEEE IEEE. 2022

  View details for DOI 10.1109/SC41404.2022.00064

  View details for Web of Science ID 000981934800039

• Compilation of Sparse Array Programming Models PROCEEDINGS OF THE ACM ON PROGRAMMING LANGUAGES-PACMPL Henry, R., Hsu, O., Yadav, R., Chou, S., Olukotun, K., Amarasinghe, S., Kjolstad, F. 2021; 5

  View details for DOI 10.1145/3485505

  View details for Web of Science ID 000731569200032