Rockwell Automation is a global technology leader focused on helping the world's manufacturers be more productive, sustainable, and agile. 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 thework we do changes the world for the better.

We are looking for a experienced AI/ML Engineer to design, build, and deploy machine learning models and intelligent systems that solve real business problems at scale. This role has a strong focus on modern generative AI - building LLM-powered applications, RAG pipelines, and agentic AI systems - alongside classical ML. You will work across the full ML/AI lifecycle, from data exploration and model development to production deployment and monitoring, partnering closely with Data Science, Product, and Platform Engineering teams. You will be reporting to Team Lead-IT
Design and build LLM-powered applications, RAG pipelines, and agentic AI workflows to automate and augment business processes.
Stay current with the latest ML/AI research (including LLMs, RAG, and agentic systems) and evaluate applicability to company use cases.
Document architectures, experiments, and decisions contribute to internal ML standards, code reviews, and knowledge sharing.
Ensure responsible AI practices - fairness, explainability, privacy, and security - are embedded throughout model development
Build and maintain robust, scalable data pipelines for feature engineering, model training, and inference (batch and real-time).
Deploy models into production using MLOps best practices - CI/CD for ML, containerization, model versioning, and automated retraining pipelines.
Monitor deployed models for drift, performance degradation, bias, and data quality issues implement retraining and alerting strategies.
Collaborate with Data Engineers to ensure clean, reliable, and well-governed data is available for model development.
Partner with Product Managers and stakeholders to translate business requirements into ML solutions and model capabilities, limitations, and trade-offs.
Conduct experiments (A/B tests,



offline evaluation) to validate model performance and business impact.
Bachelor's degree in Computer Science, Data Science, Statistics, Mathematics, Machine Learning, or a related field (or equivalent practical experience).

- 2-5+ years of hands-on experience building and deploying machine learning models in production (adjust based on seniority: Strong programming skills in Python familiarity with SQL
- Proficiency with ML/DL frameworks: 5+ years of experience with the ML lifecycle: data preprocessing, feature engineering, model training, hyperparameter tuning, evaluation, and deployment.
- Hands-on experience with MLOps tooling: MLflow, Kubeflow, SageMaker, Vertex AI, Airflow, or similar for experiment tracking, orchestration, and deployment.
- Solid understanding of statistics, probability, linear algebra, and core ML/DL algorithms.
- Experience with cloud platforms (Azure) and containerization/orchestration (Docker, Kubernetes).
- Familiarity with distributed data processing (Spark, Ray, Dask) and data pipeline/orchestration tools (Airflow, Dagster).
- 5+ years of experience with model serving and APIs (FastAPI/Flask, TorchServe, Triton Inference Server, or similar).
- Version control (Git) and CI/CD practices for ML systems.

Generative AI, LLMs & Agentic Applications (Essential)
Hands-on experience building applications powered by Large Language Models (LLMs) - e.g., Practical experience designing and implementing Retrieval-Augmented Generation (RAG) pipelines, including chunking strategies, embeddings, and retrieval evaluation.
Experience building agentic AI systems / multi-step autonomous workflows (tool use, function calling, planning, orchestration) using frameworks such as LangChain, LlamaIndex, or similar.
Working knowledge of vector databases (Pinecone,



Weaviate, FAISS, Milvus, Chroma) for semantic search and retrieval at scale.
Experience with prompt engineering, evaluation, and fine-tuning/adapting LLMs (LoRA, PEFT) for domain-specific use cases.
Understanding of the trade-offs, latency, cost, and safety considerations of deploying LLM- and agent-based applications in production.
Experience deploying and scaling multi-agent systems in production (orchestration, memory, guardrails, observability for agentic workflows).
Contributions to open-source LLM/RAG tooling, or published work on GenAI applications.
Publications, open-source contributions, or Kaggle competition experience.
Experience with real-time/streaming ML systems (Kafka, Flink, Spark Streaming).
Domain experience relevant to the business (e.g., fintech, healthcare, e-commerce, cybersecurity).
Understanding of responsible AI, model interpretability (SHAP, LIME), and privacy-preserving ML techniques.
Languages: Python, SQL
ML/DL: PyTorch, TensorFlow, scikit-learn, Hugging Face Transformers
GenAI / LLM: OpenAI, Anthropic Claude, LangChain, LlamaIndex, RAG pipelines, vector DBs (Pinecone/Weaviate/FAISS)
MLOps: MLflow, Kubeflow, SageMaker/Vertex AI/Azure ML, Airflow
Data: Spark, Pandas, Snowflake/BigQuery/Redshift
Infra: Docker, Kubernetes, AWS/GCP/Azure, Terraform
Monitoring: Prometheus/Grafana, Evidently AI, WhyLabs
Our advantages package includes:
Volunteer Paid Time Off available after 6 months of employment
Employee Assistance Program
Personalised wellbeing programmed through our OnTrack programmed
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. If you're excited about this role but your experience doesn't align perfectly with every qualification, we encourage you to apply. LI-Hybrid
#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.

📌 AI/ML Platform Engineer (Pune)
🏢 Rockwell Automation Singapore
📍 Pune