Staff Engineer, AI/ML

Rakuten Kobo Inc.; Bengaluru; Hybrid; INR 4,000,000 - 7,500,000; Full time

Job summary

Rakuten Symphony is seeking a Staff Engineer, AI/ML to drive AI-native operations for RIO across telecom OSS. You will develop time-series ML models for anomaly detection and predictive analytics, while building an AI harness and high-availability inference pipelines on top of ClickHouse/YBSQL data platforms.

You will own end-to-end ML development, provisioning, evaluation and monitoring, collaborating with data pipelines and on-call for AI services.

Qualifications

• Bachelor's or Master's degree in Computer Science, Engineering, Statistics or related field (or equivalent experience).
• 8+ years of experience in ML/AI engineering, including production deployment of models on live, high-volume data.
• Hands-on experience developing time-series ML models (anomaly detection, forecasting or classification) - deep learning and classical - with real operational impact.
• Experience building ML infrastructure: training pipelines, evaluation frameworks, feature handling, model serving and monitoring (PyTorch, scikit-learn, XGBoost/LightGBM; MLflow, Kubeflow or equivalent).
• Strong Python engineering; comfortable with large datasets via SQL (ClickHouse/YugabyteDB/PostgreSQL experience a plus).
• Solid software engineering fundamentals - version control, testing, CI/CD, containerization, Kubernetes.
• Ability to own problems end-to-end: from ambiguous operational question to productionized, monitored model.

Responsibilities

• ML Modeling: Develop and productionize multivariate time-series anomaly detection models over network KPIs.
• Forecasting and incident analysis: Build models that predict faults and correlate alarms for root-cause analysis.
• Model portfolio: Own models across the operations lifecycle from framing through production monitoring.
• AI-defined KPIs: Learn leading/lagging indicators from raw counters.
• AI Harness & Inference: Build reproducible training, experiment-tracking and hyperparameter-tuning workflows.
• Evaluation: Design offline and online evaluation frameworks for measurable impact.
• Serving: Deploy low-latency inference services with canary/rollback strategies.
• Monitoring: Own model observability and automated retraining/evaluation loops.
• LLM & agentic AI: Explore RAG over fault manuals and natural-language interfaces.
• Data Pipelines: Define feature/training-data contracts with data teams.

Skills

Python

Time-series ML

Anomaly detection

Forecasting

Root cause analysis

MLOps

Kubernetes

SQL

Education

Bachelor's or Master's in CS/Engineering/Statistics

Tools

ClickHouse

YugabyteDB

PostgreSQL

MLflow

Kubeflow

PyTorch

scikit-learn

XGBoost/LightGBM

Job description

Job Description: Job Title: Staff Engineer, AI/ML _RIO(OSS) Location: Bangalore (Hybrid)

Why should you choose us?

Rakuten Symphony is reimagining telecom, changing supply chain norms and disrupting outmoded thinking that threatens the industry’s pursuit of rapid innovation and growth. Based on proven modern infrastructure practices, its open interface platforms make it possible to launch and operate advanced mobile services in a fraction of the time and cost of conventional approaches, with no compromise to network quality or security. Rakuten Symphony has operations in Japan, the United States, Singapore, India, South Korea, Europe, and the Middle East Africa region. For more information, visit: https://symphony.rakuten.com. Building on the technology Rakuten used to launch Japan’s newest mobile network, we are taking our mobile offering global. To support our ambitions to provide an innovative cloud-native telco platform for our customers, Rakuten Symphony is looking to recruit and develop top talent from around the globe. We are looking for individuals to join our team across all functional areas of our business – from sales to engineering, support functions to product development. Let’s build the future of mobile telecommunications together!

About Rakuten Group, Inc. (TSE: 4755) is a global leader in internet services that empower individuals, communities, businesses and society. Founded in Tokyo in 1997 as an online marketplace, Rakuten has expanded to offer services in e-commerce, fintech, digital content and communications to 2 billion members around the world. The Rakuten Group has over 30,000 employees, and operations in 30 countries and regions. For more information visit https://global.rakuten.com/corp/.

About the RIO Team

RIO (Rakuten Intelligent Operations) is Rakuten Symphony's AI-first operational intelligence platform and the OSS engineering team behind it. RIO replaces fragmented OSS tooling with a single, intelligent control layer for complex, multi-vendor telecom networks: unified real-time observability across RAN, Core, Transport and Cloud; AI-driven service assurance with anomaly detection, predictive analytics and proactive fault resolution; and closed-loop, intent-based automation.

What Do We Expect From You

RIO's promise is AI-native network operations. As a Senior AI Engineer on the AI Model & Harness team, you will span two equally important halves: developing the ML models that detect anomalies, predict faults and localize root causes across telecom networks; and building the AI harness - the training, evaluation, inference and monitoring infrastructure - that takes those models from notebook to production with confidence. You will work directly on top of RIO's ClickHouse and YugabyteDB (YBSQL) data platforms, in tight partnership with the Data Management Pipelines team.

Key Responsibilities

ML Modeling (approximately 50%)

• Anomaly detection: Develop and productionize multivariate time-series anomaly detection models over network KPIs and counters to surface degraded cells and network elements before outages.
• Predictive fault & incident analysis: Build forecasting and classification models that predict faults and incidents ahead of failure, and correlate alarms across domains for root-cause analysis and noise reduction.
• Model portfolio: Own models across the operations lifecycle - KPI forecasting, event correlation, root cause localization, capacity prediction and optimization - from framing through production monitoring.
• AI-defined KPIs: Move beyond vendor KPI formulas by learning leading/lagging indicators directly from raw counters, surfacing signals humans and rule-based systems miss.

AI Harness & Inference Infrastructure (approximately 50%)

• Training & experimentation: Build reproducible training, experiment-tracking and hyperparameter-tuning workflows over large-scale telemetry datasets drawn from ClickHouse/YBSQL.
• Evaluation frameworks: Design rigorous offline and online evaluation (backtesting, precision/recall on incident ground truth, A/B and shadow deployments) so every model's operational impact is measurable.
• Serving & inference pipelines: Deploy low-latency, high-availability inference services with canary/rollback strategies; implement batch and streaming inference over near-real-time pipeline data.
• Monitoring & lifecycle: Own model observability - drift detection, data-quality gates, alerting on degradation - and automated retraining/evaluation loops.
• LLM & agentic AI: Contribute guard-railed LLM-based capabilities such as RAG over fault manuals and specifications, natural-language intent interfaces and LLM-assisted root-cause analysis.
• Partnership with data pipelines: Define feature and training-data contracts with the Data Management Pipelines team; ensure models consume and produce data through governed, well-modeled interfaces.
• Operational excellence: Participate in on-call for AI services; own the reliability of everything the harness runs in production.

Required Qualifications

• Bachelor's or Master's degree in Computer Science, Engineering, Statistics or a related field (or equivalent experience).
• 8+ years of experience in ML/AI engineering, including production deployment of models on live, high-volume data.
• Hands-on experience developing time-series ML models (anomaly detection, forecasting or classification) - deep learning and classical - with real operational impact.
• Experience building ML infrastructure: training pipelines, evaluation frameworks, feature handling, model serving and monitoring (frameworks such as PyTorch, scikit-learn, XGBoost/LightGBM; MLOps tooling such as MLflow, Kubeflow or equivalent).
• Strong Python engineering; comfortable with large datasets via SQL (ClickHouse/YugabyteDB/PostgreSQL experience a plus).
• Solid software engineering fundamentals - version control, testing, CI/CD, containerization, Kubernetes.
• Ability to own problems end-to-end: from ambiguous operational question to productionized, monitored model.

Preferred Qualifications

• Domain experience in telecom OSS, network operations, or AIOps - e.g., KPI anomaly detection, alarm correlation, root cause analysis on network telemetry.
• Experience with LLM applications in production: RAG pipelines, prompt/eval harnesses, guard-railed agent orchestration.
• Familiarity with streaming/data platforms: Kafka, Flink/Spark, ClickHouse, or YugabyteDB.
• Experience with online evaluation, shadow deployments and human-in-the-loop feedback loops.
• Public contributions to ML/AIOps research or open source (papers, repos, talks).

Rakuten Shugi Principles

• Our worldwide practices describe specific behaviours that make Rakuten unique and united across the world.
• We expect Rakuten employees to model these 5 Shugi Principles of Success.
• Always improve, always advance.
• Only be satisfied with complete success - Kaizen.
• Be passionately professional.
• Take an uncompromising approach to your work and be determined to be the best.
• Hypothesize - Practice - Validate - Shikumika.
• Use the Rakuten Cycle to success in unknown territory.
• Maximize Customer Satisfaction.
• The greatest satisfaction for workers in a service industry is to see their customers smile.
• Speed!! Speed!! Speed!!
• Always be conscious of time.
• Take charge, set clear goals, and engage your team.

Rakuten Symphony is reimagining telecom, changing supply chain norms and disrupting outmoded thinking that threatens the industry’s pursuit of rapid innovation and growth. Based on proven modern infrastructure practices, its open interface platforms make it possible to launch and operate advanced mobile services in a fraction of the time and cost of conventional approaches, with no compromise to network quality or security. Rakuten Symphony has operations in Japan, the United States, Singapore, India, South Korea, Europe, and the Middle East Africa region. For more information, visit: https://symphony.rakuten.com Building on the technology Rakuten used to launch Japan’s newest mobile network, we are taking our mobile offering global. To support our ambitions to provide an innovative cloud-native telco platform for our customers, Rakuten Symphony is looking to recruit and develop top talent from around the globe. We are looking for individuals to join our team across all functional areas of our business – from sales to engineering, support functions to product development. Let’s build the future of mobile telecommunications together!