About

I am a Deep learning compiler Enginner at NVIDIA, currently working on XLA and JAX. I have worked with LLVM and MLIR in the past and am also interested in Programming Verification and Operating Systems. I am a CSE IITH and Rice University Alumnus and I play Table Tennis.

Education

Internships and Experience

  1. Deep Learning Compiler Engineer at NVIDIA (Jan’24-Present)
  2. Compiler Intern at AMD (Aug’23 - Dec’23)
    Worked on OpenMP for AMD GPUs, by contributing to the LLVM project (flang subproject).
  3. SWE Frameworks Intern at Apple (May’23 - Aug’23)
    Developed a tool to experiment with finding cross project dependencies on macOS and iOS.
  4. System Design Engineer 2 at AMD (June’20 - Aug’22)
    Working on CPU Performance Compiler, improving coverage and speed of the AMD Compiler for HPC.
  5. Research Internship at Adobe (May’20 - July’20)
    Worked with a team of five others, on generating a visual profile of (attribute, object) combinations using compositionality and context and leverage them for enhanced image editing applications based on textual user input.
  6. Summer Internship at NTT-Advanced Technology Corp, Japan (May’19-July’19)
    Worked on developing device drivers for a LoRa safety device and an interface to provide real-time readings about the same. Also worked on setting AWS IOT button for remote control on the safety device.
  7. Teaching assistant at IIT Hyderabad
    Assisted students and teachers in various courses like Compilers, Principles of Programming Languages, Operating Systems, Computer Architecture and Software Engineering.

Research

Peer Reviewed Publications

  1. Utpal Bora, Shraiysh Vaishay, Saurabh Joshi, and Ramakrishna Upadrasta. 2021. OpenMP aware MHP Analysis for Improved Static Data-Race Detection. In Proceedings of the Seventh Annual Workshop on the LLVM Compiler Infrastructure in HPC (LLVM-HPC ’21), St. Louis, Missouri, USA. [PDF]
  2. Paridhi Maheshwari, Nihal Jain, Praneetha Vaddamanu, Dhananjay Raut, Shraiysh Vaishay, and Vishwa Vinay. 2021. Generating Compositional Color Representations from Text. In Proceedings of the 30th ACM International Conference on Information & Knowledge Management (CIKM ’21), Association for Computing Machinery, Virtual Event, Queensland, Australia, 1222–1231. DOI:https://doi.org/10.1145/3459637.3482346 [PDF]
  3. Sparsh Mittal and Shraiysh Vaishay. 2019. A survey of techniques for optimizing deep learning on GPUs. Journal of Systems Architecture 99, (2019), 101635. DOI:https://doi.org/https://doi.org/10.1016/j.sysarc.2019.101635, [PDF]

Other Research Work

  1. [Patent] Text Editing of Digital Images (P10089-US) (Submitted)
  2. [Patent] Color Representations For Textual Phrases (P10068-US) (Submitted)
  3. [Seminar] Presented a talk on “Athena: A new efficient automatic checker for security protocol analysis”. slides

Projects and Open Source Contributions

  1. Hoare-Logic Verifier
    Contract verification - interpreter for language and array logic for classroom demonstration of the application of Hoare logic on a simple programming language and verify its correctness (using lex/yacc).
  2. A Lock-free Algorithm for Parallel MCTS
    Implemented and presented the paper “S. A. Mirsoleimani., J. van den Herik. et al., “A lock-free algorithm for parallel mcts,” in Proceedings of the 10th International Conference on Agents and Artificial Intelligence - Volume 1: ICAART,, INSTICC. SciTePress, 2018.” as a project in Distributed Computing
  3. Packet Filter
    Software Defined Packet Filtering using BERT for analysis of network traffic.
  4. Leader Election in Distributed Topologies
    Using Distance-vector algorithm along with modified Bully algorithm for efficient leader election in connected topologies.
  5. Crowds(Prism)
    Implemented the Crowds protocol in Prism model checker and studied the the factors affecting probability of detection of initiator in an Anonymity System. We also suggested an algorithm for placing corrupt nodes to optimally detect the source of a message.
  6. Code-Compliance
    Worked on a static analysis of C/C++ programs (Code Compliance) using Clang/LLVM. Verified a subset of rules from certification standards like CERT, MISRA etc.
  7. Distant Supervision for Relation Extraction without Labeled data
    We utilise embeddings from BERT and fine tune it using manifold mixup to achieve distantly supervised relation extraction as a course project on Data Mining.
  8. COOL - Classroom Object Oriented Language
    Worked on writing Lexer, Parser, Semantic Analyzer and LLVM IR code generator for COOL as a part of classroom project (The code generator is a private repository).
  9. LLVM/MLIR Contributions
    Worked on translation of LLVM Dialect in MLIR to LLVM IR facilitating wider code and target coverage for MLIR.
  10. HasQL
    Built query handlers (back-end) for a local database system with a simple query language, using functional programming in Haskell.
  11. GoTorrent
    Built messages and handled concurrent downloads of file fragments for a BitTorrent client in Go. Features include concurrent download from peers, concurrent download of multiple torrents and handling of abrupt termination of download.

Skills

  1. Compilers and Programming Languages
    Experienced with Compiler Design. Worked with LLVM, MLIR, Clang and Flang. Designed and worked on a language for verifying simple programs using hoare logic. Also worked on COOL Compiler.
  2. Operating Systems, Distributed Systems, Parallel and Concurrent Programming
    Familiar with various problems and algorithms for distributed and parallel computing. Experience developing device drivers for linux.
  3. Languages
    Familiar with a wide range of languages including C/C++, Python, Go and Java. Well versed in functional programming in Haskell.
  4. Git
    Strong knowledge of project management in teams, experienced with open source contributions.
  5. Systems engineering
    Server, network and storage administration, virtualization and containers, and experience with AWS.

Extracurricular and Voluntary work

  1. Member of NSO(National Sports Organisation) at IIT Hyderabad (Table-Tennis)
  2. Completed three years of certified classroom program of Abacus by BrainOBrain (2011-14)
  3. Spreading awareness about vaccination among children in rural areas