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
