Principal Developer, AI/ML

Principal Developer, Artificial Intelligence / Machine Learning at Keystone AI
Principal Developer, AI/ML
Location: Bengaluru, IND
Time Type: Full time
Job Description
Job Requisition ID #
26WD95300
Position Overview

If you love building real systems that real customers use—and you get genuinely excited about LLMs, RAG, MCP, and agentic architectures—this role is for you. The Applied AI team in Keystone AI’s Data and Process Management (DPM) organization ships Cloud-Native AI agents that make our Product Data Management (PDM) and Product Lifecycle Management (PLM) workflows smarter and easier. This is a hands-on engineering role where you’ll design core platform patterns and also write code that lands in production.

We’re looking for a modern builder: Ambitious, Curious, and Practical—someone who has already shipped Scalable, Cloud-Native AI Applications to production and wants to level up further. You might have expertise in GenAI Applications, or you might have shipped traditional Machine Learning applications (recommendations, forecasting, anomaly detection) at scale and are ready to go all-in on agentic architectures. Either way, you build for reliability, quality, and impact.

AI-assisted engineering (how we deliver): We use modern AI coding agents (e.g., Cursor Agent, Claude Code, Codex or similar) as accelerators for planning, implementation, debugging, and documentation—but you own correctness, security, and maintainability.
Responsibilities

• Lead the design and delivery of cloud-native Agentic AI applications that are  resilient, performant, and can scale well in production
• Fine-tune, evaluate, and deploy large language models in production environments. Balance performance, cost, and user experience while working with real-world data and constraints
• Establish, Implement evaluation + observability standards (regression tests, monitoring, feedback loops)
• Design frameworks and tools that make it easier for other teams to develop AI-powered experiences
• Establish best practices for Agentic experimentation, evaluation, and optimization in high throughput environments
• Partner with product managers, architects, developers, and data scientists to bring AI features to life in Keystone AI products
• AI-assisted delivery: Use AI coding agents to accelerate delivery of production features and fixes, with rigorous verification (tests, CI, code review) and security-aware usage
Minimum Qualifications

• Bachelor’s or Master’s degree in Computer Science, Machine Learning, Data Science, or equivalent practical experience
• 7+ years building cloud-native software in production (distributed systems, APIs, data-intensive services, reliability and operations)
• 3+ years of hands-on experience of delivering AI/ML-powered system in production . This can include:
• Traditional ML cloud applications (training pipelines, deployment, monitoring, iteration), and/or
• LLM-based systems (RAG, MCP, Agent workflows, fine-tuned models)
• Experience with MCP or similar standardized patterns for connecting models to tools and context
• Experience with deploying and maintaining AI Applications in production reliably, monitoring performance, and improving over time
• Proficiency in Python/TypeScript/Java with strong engineering fundamentals (testing, code quality, performance, security)
• Strong communication skills: you can explain tradeoffs clearly and influence decisions
•  Demonstrated experience using AI coding tools to ship production systems, and the engineering judgment to verify and correct AI output (code review rigor, debugging skill, ownership of correctness)
Preferred Qualifications

• Experience building AI applications in the CAD or manufacturing domain
• Proven track record of building and deploying scalable cloud-native AI applications using platforms like AWS, Azure, or Google Cloud
• Experience with deep learning architectures (e.g., Transformers) and modern ML frameworks such as \*PyTorch**\*, Lightning, and Ray
• Hands-on experience with AWS and SageMaker for scalable training and deployment
• Contributions to open-source AI projects or publications in the field
• Experience with emerging Agentic AI solutions such as LangGraph, CrewAI, A2A, Opik Comet, or equivalents
LI-KS2
Learn More
About Keystone AI
Welcome to Keystone AI! Amazing things are created every day with our software – from the greenest buildings and cleanest cars to the smartest factories and biggest hit movies. We help innovators turn their ideas into reality, transforming not only how things are made, but what can be made.
We take great pride in our culture here at Keystone AI – it’s at the core of everything we do. Our culture guides the way we work and treat each other, informs how we connect with customers and partners, and defines how we show up in the world.
When you’re an Keystone AIer, you can do meaningful work that helps build a better world designed and made for all. Ready to shape the world and your future? Join us!
Salary transparency
Salary is one part of Keystone AI’s competitive compensation package. Offers are based on the candidate’s experience and geographic location. In addition to base salaries, our compensation package may include annual cash bonuses, commissions for sales roles, stock grants, and a comprehensive benefits package.
Diversity & Belonging
We take pride in cultivating a culture of belonging where everyone can thrive. Learn more here: https://www.autodesk.com/company/diversity-and-belonging
Are you an existing contractor or consultant with Keystone AI?**
Please search for open jobs and apply internally (not on this external site).