AI/ML Engineer
Rockwell Automation6 days ago
Pune, India or Bengaluru, IndiaMid Level / Senior
Responsibilities
- Design and build LLM-powered applications, RAG pipelines, and agentic AI workflows.
- Develop and maintain scalable data pipelines for feature engineering, training, and batch or real-time inference.
- Deploy, version, monitor, and retrain production models using MLOps practices.
- Monitor model drift, performance, bias, and data quality, implementing alerting and retraining strategies.
- Collaborate with Data Engineers, Data Scientists, Product Managers, Platform Engineering, and stakeholders on ML solutions.
- Conduct A/B tests and offline evaluations to measure model performance and business impact.
- Optimize models and inference systems for latency, throughput, cost, and production efficiency.
- Apply responsible AI practices involving fairness, explainability, privacy, security, and safety.
- Document architectures and experiments, participate in code reviews, and contribute to ML standards and knowledge sharing.
Requirements
- 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, with the posting also specifying 5+ years for the ML lifecycle and model serving areas.
- Strong Python programming skills and familiarity with SQL.
- Experience with TensorFlow, PyTorch, scikit-learn, XGBoost or LightGBM, and core machine learning and deep learning methods.
- Experience with MLflow, Kubeflow, SageMaker, Vertex AI, Airflow, or similar MLOps tools.
- Experience with Azure, Docker, Kubernetes, distributed data processing, and data pipeline orchestration.
- Experience with model serving and APIs such as FastAPI, Flask, TorchServe, or Triton Inference Server.
- Hands-on experience building applications with LLM APIs such as OpenAI, Anthropic Claude, Llama, Mistral, or Gemini.
- Practical experience implementing RAG pipelines, embeddings, retrieval evaluation, prompt engineering, and LLM evaluation or adaptation using techniques such as LoRA or PEFT.
- Experience building agentic AI systems and multi-step workflows using tools or frameworks such as LangChain or LlamaIndex.
- Working knowledge of vector databases including Pinecone, Weaviate, FAISS, Milvus, or Chroma.
- Understanding of statistics, probability, linear algebra, model optimization, model serving, and production trade-offs.
- Preferred qualifications include multi-agent production experience, open-source or publication experience, Kaggle experience, streaming ML experience, domain experience, and knowledge of interpretability and privacy-preserving ML.
Benefits
- Comprehensive mindfulness programming with a premium Calm membership.
- Volunteer paid time off available after six months of employment.
- Employee volunteer and donation matching program.
- Employee Assistance Program.
- Personalized wellbeing programming through the OnTrack program.
- On-demand digital courses for professional development.
- Hybrid work arrangement with expected Rockwell-location work on Mondays, Tuesdays, and Thursdays unless business obligations require otherwise.
Tech Stack
Amazon RedshiftApache AirflowApache FlinkApache KafkaApache SparkAWSAzureDockerFastAPIFlaskGitGoogle BigQueryGoogle Cloud PlatformGrafanaHugging Face TransformersKubernetesLightGBMMLflowPandasPrometheusPythonPyTorchscikit-learnSnowflakeSQLTensorFlowTerraformXGBoost
Categories
About Rockwell Automation
Rockwell Automation builds industrial control hardware and factory software for manufacturers, and sells services and systems to design, operate, and maintain automated plants. Its Allen-Bradley controllers, drives, and safety products pair with FactoryTalk software and MES to support discrete and process industries. Founded in 1903 and headquartered in Milwaukee, it is a public company on the NYSE (ROK) with global sales, partners, and OEM customers.