Senior Staff ML Researcher

Client is seeking machine learning researchers to join our Algo Team. We’re looking for someone to invent, design, and implement efficient algorithms that will be used to optimize Large Language Model inference on DNN Accelerators we develop. You would be part of a close-knit team of mathematicians, ML researchers, and ML engineers who create and apply advanced algorithmic and numerical techniques to the most cutting-edge and high-impact research in the overlap of mathematics, ML, and modern LLM applications.

Hybrid, working on-site at our Bangalore, India, offices 3 days per week

• Strong mathematical skills—MSc or PhD in math, CS, statistics, physics, or a related STEM field

• Experience in Python and OOP code design

• Experience with transformer architecture is advantageous but not mandatory