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
-
- Master of Computer Science, 2023
- Rice University
-
- BTech (Hons.) in Computer Science and Engineering, 2021
- IIT Hyderabad
-
- Intermediate
- 11th and 12th Standard - Telangana State Board of Intermediate Education, Government of Telangana
-
- Highschool
- 10th Standard - I.C.S.E. Board
Internships and Experience
- Deep Learning Compiler Engineer at NVIDIA (Jan’24-Present)
-
- Compiler Intern at AMD (Aug’23 - Dec’23)
- Worked on OpenMP for AMD GPUs, by contributing to the LLVM project (flang subproject).
-
- SWE Frameworks Intern at Apple (May’23 - Aug’23)
- Developed a tool to experiment with finding cross project dependencies on macOS and iOS.
-
- 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.
-
- 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.
-
- 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.
-
- 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
- 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]
- 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]
- 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
- [Patent] Text Editing of Digital Images (P10089-US) (Submitted)
- [Patent] Color Representations For Textual Phrases (P10068-US) (Submitted)
- [Seminar] Presented a talk on “Athena: A new efficient automatic checker for security protocol analysis”. slides
Projects and Open Source Contributions
-
- 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).
-
- 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
-
- Packet Filter
- Software Defined Packet Filtering using BERT for analysis of network traffic.
-
- Leader Election in Distributed Topologies
- Using Distance-vector algorithm along with modified Bully algorithm for efficient leader election in connected topologies.
-
- 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.
-
- 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.
-
- 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.
-
- 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).
-
- LLVM/MLIR Contributions
- Worked on translation of LLVM Dialect in MLIR to LLVM IR facilitating wider code and target coverage for MLIR.
-
- HasQL
- Built query handlers (back-end) for a local database system with a simple query language, using functional programming in Haskell.
-
- 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
-
- 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.
-
- 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.
-
- Languages
- Familiar with a wide range of languages including C/C++, Python, Go and Java. Well versed in functional programming in Haskell.
-
- Git
- Strong knowledge of project management in teams, experienced with open source contributions.
-
- Systems engineering
- Server, network and storage administration, virtualization and containers, and experience with AWS.
Extracurricular and Voluntary work
- Member of NSO(National Sports Organisation) at IIT Hyderabad (Table-Tennis)
- Completed three years of certified classroom program of Abacus by BrainOBrain (2011-14)
- Spreading awareness about vaccination among children in rural areas