AI/ML Tech Lead

Rockwell Automation is a global technology leader focused on helping the world’s manufacturers be more productive, sustainable, and agile. With more than 28,000 employees who make the world better every day, we know we have something special. Behind our customers - amazing companies that help feed the world, provide life-saving medicine on a global scale, and focus on clean water and green mobility - our people are energized problem solvers that take pride in how the work we do changes the world for the better.

We welcome all makers, forward thinkers, and problem solvers who are looking for a place to do their best work. And if that’s you we would love to have you join us!

Job Description

AI/ML Tech Lead  

Pune/Noida/Bangalore, India 

As an AI AI/ML Tech Lead, you will lead the design, development, and deployment of enterprise-grade artificial intelligence solutions that drive innovation and deliver measurable business value. You will report to the Technical Leader of Enterprise Architecture and work in a hybrid capacity from our Pune, India office. In this role, you will collaborate with cross-functional teams—including data scientists, engineers, and business leaders—to ensure AI systems are scalable, secure, and ethically aligned with organizational goals. 

Your Responsibilities: 

• Architect and implement AI solutions: Design and oversee end-to-end AI architectures that integrate seamlessly with existing IT and data infrastructure, ensuring scalability, performance, and maintainability. 

• Lead cross-functional delivery teams: Guide data scientists, engineers, and business stakeholders through the full AI solution lifecycle—from ideation and prototyping to production deployment and monitoring. 

• Evaluate and recommend technologies: Assess AI/ML platforms, frameworks, and tools (e.g., TensorFlow, PyTorch, cloud AI services) to ensure alignment with business needs and technical feasibility. 

• Establish best practices: Define standards for model development, testing, deployment, and lifecycle management, ensuring compliance with ethical AI principles and data privacy regulations. 

• Mentor and evangelize: Provide technical leadership and mentorship to  AI Engineers and data professionals, while promoting AI adoption and architectural vision across the organization. 

The Essentials – You Will Have: 

• A Bachelor's or Master's degree in Computer Science, Artificial Intelligence, Data Science, or a related field. 

• 5+ years of experience designing and implementing AI solutions in production environments. 

• Proficiency with AI/ML frameworks and tools such as TensorFlow, PyTorch, Keras, Scikit-learn, and cloud-based AI services (e.g., AWS SageMaker, Azure ML). 

• Strong programming skills in Python, R, Java, or similar languages. 

• Deep understanding of data structures, algorithms, and software engineering best practices. 

• Demonstrated experience leading complex AI projects with cross-functional teams. 

• Proven track record of delivering AI solutions that drive business outcomes. 

• Experience with ethical AI practices and compliance with data protection regulations. 

The Preferred – You Might Also Have: 

• Experience deploying AI solutions on cloud platforms (AWS, Azure, GCP) in hybrid or multi-cloud environments. 

• Familiarity with MLOps tools and practices for continuous integration, deployment, and monitoring of AI models. 

• Strong problem-solving skills and the ability to translate business requirements into scalable technical solutions. 

• Experience mentoring and developing talent within AI or data science teams. 

What We Offer: 

Our benefits package includes … 

• Comprehensive mindfulness programs with a premium membership to Calm 

• Volunteer Paid Time off available after 6 months of employment for eligible employees 

• Company volunteer and donation matching program – Your volunteer hours or personal cash donations to an eligible charity can be matched with a charitable donation. 

• Employee Assistance Program 

• Personalized wellbeing programs through our OnTrack program 

• On-demand digital course library for professional development 

... and other local benefits! 

At Rockwell Automation we are dedicated to building a diverse, inclusive and authentic workplace, so if you're excited about this role but your experience doesn't align perfectly with every qualification in the job description, we encourage you to apply anyway. You may be just the right person for this or other roles. 

Rockwell Automation’s hybrid policy aligns that employees are expected to work at a Rockwell location at least Mondays, Tuesdays, and Thursdays unless they have a business obligation out of the office.