Dec 2022 – Mar 2024 · 1 yr 3 mos · Bengaluru, Karnataka, India · Hybrid

Feb 2022 – Nov 2022 · 9 mos · Remote

• Worked as a Research Engineer at Lightning AI. Most of my work includes designing, drafting, and implementing features for the libraries in the Lightning ecosystem.
• Lead: Lightning Flash
• Maintainer: PyTorch Lightning (Lightning Trainer)

May 2021 – Feb 2022 · 9 mos

• (Contributor - PyTorch) Worked with the PyTorch team at Quansight as a software developer. Most of my tasks include fixing bugs, adding features and implementing functions and/or modules in PyTorch.

Apr 2020 – Jan 2021 · 9 mos · Remote

• Worked on deep learning, speech recognition using Kaldi and optimizing DL models using TensorRT.

Jan 2020 – Apr 2020 · 3 mos · Samta Clara, California

• Worked as a SDE Intern in the PyTorch DL Frameworks Team at NVIDIA (HQ: Santa Clara, US). My work was mostly focused on enabling type promotion for Unary Ops for both CUDA and CPU devices. The target was to prevent extra kernel launches by doing the type promotion inside the TensorIterator of PyTorch. The tasks also included fixing bugs in PyTorch (CUDA related), and adding explicit dtype argument to Unary Ops (which will help easy type promotion for our AMP Pull Request).
• The work was done under mentorship of Michael Carilli (Senior Software Developer, NVIDIA), Piotr Bialecki (Senior Software Developer, NVIDIA) and my manager Christian Sarofeen. Most of the work was also done with the help of awesome people at Facebook (PyTorch Team): Michael Ruberry, Natalia, Will Feng and more.
• The work heavily used C++, CUDA and Python (to write tests).
• I also contributed to PyTorch actively during my internship (and after as well). Please take a look at the PRs specific to my time at NVIDIA:
• 1. Enabling Type Promotion: https://github.com/pytorch/pytorch/pull/33322
• 2. Adding dtype arguments to the Unary Ops for explicit type promotion: https://github.com/pytorch/pytorch/pull/33063
• 3. Fix torch.prod: https://github.com/pytorch/pytorch/pull/32831
• Please take a look at: https://github.com/pytorch/pytorch/pulls/krshrimali if you want to look at my contributions to PyTorch.

May 2019 – Jul 2019 · 2 mos · NTU Singapore

• Joined as a Visiting Researcher at ROSE Labs, NTU Singapore, under supervision of Professor Alex Kot (Director, ROSE Labs) and Dr. Dennis Sng (Deputy Director, ROSE Labs).
• Worked on License Plate Detection and Recognition using YOLOv3 based network. Target on detecting and recognizing characters for Double License Plates and on realtime dataset.
• Work involved multiprocessing to generate synthetic dataset for training custom ALPR.

Nov 2018 – Mar 2019 · 4 mos · IIIT Naya Raipur

• Coordinated clubs running under SAC:
• 1. AIML (facebook.com/aiml.IIITNR/)
• 2. TSoC (The Society of Coders)
• Technical Coordinator of Technical and Cultural Fest - TechNovate 2019.

Mar 2018 – May 2019 · 1 yr 2 mos · California

• Started working as an intern at Big Vision LLC under Dr. Satya Mallick, founder of learnopencv.com, TAAZ and Big Vision LLC.
• Field: Computer Vision and Machine Learning.
• Technical blogs written:
• 1. Image Quality Assessment (using BRISQUE, a No-Reference Technqiue): https://www.learnopencv.com/image-quality-assessment-brisque/
• 2. Face Averaging (Application-based): https://www.learnopencv.com/average-faces-of-fifa-world-cup-2018/
• 3. SVM using Scikit-Learn (in Python): https://www.learnopencv.com/svm-using-scikit-learn-in-python/
• 4. Convex Hull using OpenCV in C++ and Python: https://www.learnopencv.com/convex-hull-using-opencv-in-python-and-c/
• 5. Code for PyTorch Image Classification (Transfer Learning) using PyTorch: https://www.learnopencv.com/image-classification-using-transfer-learning-in-pytorch/
• Programming Languages used: C#, C++, Python, JavaScript

Mar 2018 – Mar 2018 · 0 mo · IIIT Naya Raipur

• Headed Machine Learning and Data Science event, along with Krutika Bapat (B.Tech CSE Undergraduate, IIIT Naya Raipur).

Feb 2018 – Feb 2018 · 0 mo · IIIT Naya Raipur

• Gave lecture on importance of Computer Science Education, to the students of high-school, of Chhattisgarh.

Dec 2017 – Jan 2018 · 1 mo · Gwalior

• Internship on Multi-Agent Systems, Deep Learning Models at IIIT Gwalior, under Professor Dr. Anupam Shukla. Also focused to learn Hybrid Models developed in the labs of IIIT Gwalior.
• Algorithms like A* Search, Probabilistic Distance, and Genetic were modeled, studied and learned. Project was done and implemented using Python, having multiple robots and multiple goals currently in a static environment, future goal being - for dynamic environment.

Nov 2016 – Dec 2016 · 1 mo · Raipur

• Technical Content Writer.
• Blog post: http://www.geekwinky.in/2016/11/handling-linux-files-in-windows.html

Jun 2013 – Jun 2017 · 4 yrs

• Newsletter Programme Leader, Designer.
• OpenStudy, an online educational platform for the students to get their doubts cleared, interactively. It has quite a lot users, and is now occupied by Brainly. The task, as an ambassador, was to guide the new users towards the rules, the starting steps etc.