Backend Engineer - AI/ML Deployment (Open-Source Model Pipelines &

Reinforcement Learning)

Location: Navi Mumbai, India

Job Type: Full-Time

Job Summary

Jio Reality Labs is seeking an experienced Backend Engineer to design, deploy, and manage

scalable AI/ML pipelines that support our machine learning applications, serving millions of

users. This role will focus on building robust pipelines for deploying open-source models and

implementing reinforcement learning solutions. If you have a passion for advanced AI

technologies and expertise in deploying high-performance ML systems, we'd love to hear from

you!

Responsibilities

AI/ML Pipeline Development and Deployment: Design and implement end-to-end pipelines

for deploying open-source machine learning models with a focus on scalability, performance,

and monitoring. Collaborate with data science teams to build reinforcement learning pipelines,

managing real-time policy updates, feedback loops, and continuous model improvement.

Open-Source Model Management: Deploy and optimize models using frameworks like

TensorFlow, PyTorch, and ONNX, focusing on efficient resource usage and scalable deployment.

Containerize models with Docker and orchestrate deployments with Kubernetes to ensure

consistency and reliability across environments.

Reinforcement Learning (RL) Implementation: Architect and manage RL pipelines that handle

real-time data ingestion, policy updates, and action feedback loops to enhance decision-making

in AI-driven applications. Support large-scale training and deployment of RL agents, leveraging

distributed systems to ensure timely updates and optimized performance.

Backend Engineering & Cloud Infrastructure: Use cloud platforms (AWS, GCP, Azure) to build

scalable, high-performance infrastructure that supports our ML workloads. Implement

automated scaling, load balancing, and optimization to handle millions of user requests with low

latency.

MLOps and CI/CD Automation: Develop CI/CD pipelines to streamline model deployment,

monitoring, and updates in production environments. Use monitoring tools (Prometheus,

Grafana) to track model performance, system health, and ensure rapid troubleshooting.

Data Security and Compliance: Apply best practices in data security and compliance, including

encryption and user data anonymization, to adhere to industry standards and protect customer

data.

Required Skills And Qualifications

• Bachelor's or Master's Degree in Computer Science, Engineering, or a related field. 3+ years of experience in backend engineering with expertise in deploying scalable AI/ML

pipelines.

Proficiency in deploying open-source models with frameworks like TensorFlow, PyTorch, or

ONNX.

Experience implementing reinforcement learning pipelines, including real-time policy updates

and feedback mechanisms.

• Strong knowledge of containerization and orchestration (Docker, Kubernetes). Proficiency in cloud platforms (AWS, GCP, or Azure) for deploying and scaling AI/ML

infrastructure.

• CI/CD and MLOps expertise, including experience with Git, Jenkins, MLflow, and Kubeflow. Data security knowledge for compliance and secure API management.

Preferred Skills

Experience with distributed computing frameworks (Apache Spark, Ray) for parallelizing data

processing and model training.

• Knowledge of edge computing for optimizing AI inference closer to the end-user. Familiarity with data lake and data warehouse solutions (BigQuery, Redshift, Snowflake).

What We Offer

An opportunity to work with cutting-edge AI/ML technologies and contribute to innovative

projects.

• A collaborative and dynamic work environment focused on continuous learning and growth. Flexible work hours and competitive salary and benefits package.

If you're ready to bring AI to scale at Jio Reality Labs, apply with your resume and a cover letter

detailing your experience with open-source model deployment and reinforcement learning!

Skills: python,reinforcement learning,jenkins,kubernetes,mlops,ci/cd,azure,ai/ml pipelines