Top Machine Learning Jobs in India

Lead engineer responsible for end-to-end bring-up of ML models and frameworks on Cerebras CSX systems. Work across model translation, graph lowering, compiler optimizations, runtime integration, and performance tuning. Debug performance and correctness issues across model code, compiler IRs, runtime behavior, and hardware utilization. Prototype tooling and API improvements to accelerate future bring-ups.

Design and build large-scale personalized recommendation systems using state-of-the-art ML (deep learning, transformers, bandits, LLMs, causal inference). Run and analyze A/B tests, collaborate cross-functionally to translate requirements, scale ML infrastructure and data pipelines, and mentor junior engineers.

The role involves leading AI ML initiatives, managing product development, ensuring adherence to software best practices, and overseeing project delivery in a global context.

Manage a team of AI Engineers and Data Scientists to deliver AI-ML solutions, ensuring quality, scalability, and alignment with business goals.

The AI/ML Consultant will design automation strategies for Order-to-Cash workflows, identifying improvement areas and delivering a comprehensive roadmap for implementation.