Abhijit Balaji

AI Researcher

Seattle, Washington, United States5 yrs 5 mos experience
AI EnabledAI ML Practitioner

Key Highlights

  • Expert in Machine Learning and AI technologies.
  • Led development of innovative AI solutions at top tech companies.
  • Proven track record in mentoring and team management.
Stackforce AI infers this person is a Machine Learning Engineer with expertise in AI solutions for consumer electronics and digital media.

Contact

Skills

Core Skills

Machine LearningArtificial IntelligenceComputer Vision

Other Skills

Algorithm DevelopmentAlgorithmsC++Cloud ComputingData AnalysisData ScienceDeep LearningDockerEdge ComputingGitGithubGoogle Cloud Platform (GCP)Image ProcessingKerasMachine Vision

About

An AI enthusiast with a passion for deep learning, computer vision and natural language processing.

Experience

Rain

Senior Machine Learning Engineer - Tech Lead

May 2023Jun 2025 · 2 yrs 1 mo · United States · Remote

  • ML Tech lead and creator of TorchMX: https://github.com/rain-neuromorphics/torchmx
  • Designed and developed Rain’s LLM quantization framework - TorchMX enabling seamless compilation and quantization of any PyTorch-based LLM from HuggingFace to Rain’s proprietary hardware.
  • Built and deployed a secure, GitHub Copilot-style coding assistant using CodeLlama and vLLM, integrated into VSCode, operating fully on-premise to maintain data privacy across the company.
  • Mentored and managed ML interns, fostering a research-driven high-performance culture.
Google Cloud Platform (GCP)TensorFlowEdge ComputingComputer VisionGitPython (Programming Language)+4

Google

Software Engineer - On Device Machine Learning

Jan 2022Mar 2023 · 1 yr 2 mos · Mountain View, California, United States

  • Perception and On Device Machine Learning
  • Created a deep learning model for the Google Pixel tablet that tracks a human's position in real time.
  • Created a generative model that auto corrects a person's gaze in real time.
  • Engineered an on device system that facilitates model updates for deployed ML models in Nest doorbells and Nest cameras.
TensorFlowOpenCVData AnalysisDeep LearningComputer VisionCloud Computing+7

Adobe

Machine Learning Engineer

May 2021Aug 2021 · 3 mos · San Jose, California, United States

  • Worked on "Fine Grained Multi-Modal Representation Learning".
  • Built a fine-grained text to image retrieval system for Adobe Stock which is significantly better in fetching relevant images for semantically rich queries.
  • Created a novel training methodology which improves the performance of CLIP (OPENAI) model using Denotation Graphs.
TensorFlowOpenCVData AnalysisDeep LearningPythonC+++6

Samsung electronics america

Machine Learning Research Engineer

Aug 2020Dec 2020 · 4 mos · Mountain View, California, United States

  • Worked with the Samsung Research America's Think Tank Team. Where I:
  • Conducted research on computational photography.
  • Developed deep learning models for portrait image relighting.
  • Programmed a Python library for interacting with AWS S3 for efficient storing and retrieval of data while conducting large scale machine learning experiments.
TensorFlowOpenCVData AnalysisDeep LearningPythonC+++6

The university of texas at dallas

Research Assistant

May 2020Aug 2020 · 3 mos

  • Created forecasting models that optimized the bidding plans for an energy company.
TensorFlowOpenCVData AnalysisDeep LearningPythonC+++6

Straive

Data Scientist

Aug 2018Jun 2019 · 10 mos · Greater Chennai Area

  • Built an AI system that describes chart images which in turn enhances the online accessibility for the visually impaired.
  • Designed and deployed a NLP model for Bio-Medical data that enhanced the search capability of the research paper search engine.
  • Led a team that devised an AI proof-reader for books, journals and documents that improved the productivity of human copy editors by over 500% within the company.
TensorFlowOpenCVData AnalysisDeep LearningPythonC+++6

Endotherm fluids (india) private limited

Data Science Engineer

May 2017Jun 2018 · 1 yr 1 mo · Greater Chennai Area

  • Developed and deployed a CNN based machine vision system that detects surface defects in metal components.
  • Managed and oversaw a team that engineered an entire robotic conveyor system that uses the above AI machine vision system to conduct quality inspection.
  • Launched a startup company called JIDOKA Technologies that provides AI based custom machine vision solutions based on the above research.
TensorFlowOpenCVData AnalysisDeep LearningPythonC+++6

Education

The University of Texas at Dallas

Master of Science - MS — Computer Science

Vellore Institute of Technology

Bachelor of Technology - BTech — Mechanical Engineering

Stackforce found 100+ more professionals with Machine Learning & Artificial Intelligence

Explore similar profiles based on matching skills and experience