Keerthan Shagrithaya

AI Researcher

Bengaluru, Karnataka, India6 yrs 4 mos experience

Highly StableAI Enabled

Key Highlights

Expert in Machine Learning and MLOps.
Developed high-performance deep learning libraries.
Researcher in NLP and Reinforcement Learning.

Stackforce AI infers this person is a Machine Learning Engineer with expertise in Food-Tech and Deep Learning.

Contact

Skills

Core Skills

Machine LearningAmazon Web Services (aws)Natural Language Processing (nlp)Reinforcement LearningC++C (programming Language)Deep Learning

Other Skills

MLflowHigh Performance Computing (HPC)Python (Programming Language)SciPyGenerative AILarge Language Models (LLM)JavaAndroidHTMLMicrosoft Office

About

I am a driven software engineer passionate about advances in Machine Learning, MLOps and High Performance Computing. I've worked across the board, from fast neural network inference at AMD, to research in Reinforcement Learning and Natural Language Processing at IISc, to my most recent role in applying ML with hyperspectral imaging to further food-tech. I'm always excited to learn more and have interesting conversations. Let's chat!

Experience

6 yrs 4 mos

Total Experience

2 yrs 1 mo

Average Tenure

Current Experience

Imagoai

Machine Learning Lead

Oct 2024 – Apr 2025 · 6 mos · Remote

Built infrastructure for automated training, deployment and monitoring of machine learning models for mycotoxin detection in crops via hyperspectral image processing using AWS and MLFlow
Instrumental in extending company’s portfolio to 6 new offerings across multiple varieties of toxins and crop types
Delivered consistent, data-wide model reliability through targeted robustness strategies

Machine LearningAmazon Web Services (AWS)

Indian institute of science (iisc)

2 roles

Ph.D. Student

Aug 2021 – Aug 2024 · 3 yrs

Specialisation: Natural Language Processing (NLP)
Researched efficient coreference resolution using a single encoder pass.
Researched code generation using LLMs for competitive coding challenges.
Teaching Assistant for Linear Algebra & Probability
Courses: Deep Learning for NLP, Reinforcement Learning, Machine Learning, Computational Models of Optimisation

Natural Language Processing (NLP)Reinforcement Learning

Research Assistant

Aug 2020 – Jul 2021 · 11 mos

Developed a novel form of Q-learning to observe non-Markovian constraints in reinforcement learning.
Designed a preference-based inverse RL algorithm to learn costs and rewards separately.
Collaborated with an early stage startup to create an object detection solution for tracking football players using multi-agent RL.

Reinforcement LearningMachine Learning

Amd

2 roles

Software System Designer 2

Jul 2020 – Jul 2020 · 0 mo

Software System Designer

Jul 2018 – Jun 2020 · 1 yr 11 mos

Among founding engineers of ZenDNN, a high performance library for deep learning inference.
Developed backend convolution kernels tailored for Tensorflow on AMD CPUs that beat the then state of the art by 15%.
Designed graph optimisations via innovative fusion techniques to facilitate reduced kernel calls and better memory management.
Collaborated on ideating techniques in neural network pruning and quantization that led to the filing of 2 patents.

C++C (Programming Language)