Machine Learning Engineer - Agentic AI & Reinforcement Learning

Job Title: Machine Learning Engineer - Agentic AI & Reinforcement Learning Company: Viridien Job Category: Machine Learning Roles & Responsibilities: As a member of our AI Lab, you'll collaborate with senior machine learning engineers, software developers, HPC specialists and scientific domain experts across Viridien. Learn more about the team on our website , in a recent blog post featuring one of our senior machine learning engineers and from our Voices of Viridien. * Designing and developing single\-agent and multi\-agent systems. * Implementing and evaluating capabilities such as reasoning, planning, task decomposition, memory, tool use and agent collaboration. * Integrating LLMs, vision\-language models, retrieval systems and domain\-specific tools. * Experimenting with fine\-tuning and adapting foundation models using supervised learning, preference optimisation and reinforcement learning where appropriate. * Developing training environments, reward functions, graders and feedback mechanisms. * Designing evaluations, guardrails and monitoring approaches for agent reliability. Helping turn successful prototypes into APIs, services and reusable software components. * Learning and development We encourage and actively support a strong sense of community, through volunteering and various company initiatives. As a company we are also committed to reducing our environmental impact, improving energy efficiency and accelerating sustainable solutions. Work Location: Maharashtra, India Availability: Full time

* Degree in computer science, artificial intelligence, machine learning, engineering or a related discipline, or equivalent practical experience. * Good Python programming skills and an understanding of software engineering practices. * Experience building machine\-learning, generative\-AI or LLM applications, gained through employment, industry or research internships, applied research or significant academic projects. * Experience with PyTorch, JAX, TensorFlow or a similar framework. * Familiarity with core agentic AI concepts such as tool calling, planning, memory and workflow orchestration. * An understanding of reinforcement learning, preference optimisation or sequential decision\-making. Some experience evaluating models or implementing ideas from machine\-learning research. * * Familiarity with multi\-agent systems using LangGraph, AutoGen, smolagents or similar frameworks. * Some exposure to LLM fine\-tuning or preference and reinforcement\-learning approaches, such as DPO, GRPO, RLHF or RLAIF. * Experience designing reward functions, graders or agent training environments. * Familiarity with Model Context Protocol or other approaches to tool integration. * Exposure to training or deploying open\-source language or vision\-language models using GPUs, including distributed training. Experience applying AI to scientific, engineering, energy or geoscience workflows. * At Viridien, we are dedicated to providing the needs of our employees; and we know every employee has unique personal goals, family circumstances, and preferences. That is why we offer a wide array of benefits to promote and enhance your total well\-being: At Viridien, we are committed to delivering a respectful, inclusive, and transparent recruitment experience. Due to the high volume of applications we receive, we may not be able to provide individual feedback to every applicant. Only candidates whose qualifications closely match the role criteria will be contacted for an interview. We do, however, aim to share personalized feedback with those who progress to the first round of interviews and beyond. We are also dedicated to ensuring that our hiring process accessible to all. If you require any reasonable adjustments to fully participate in the application or interview stages, please don’t hesitate to contact your recruiter directly. Key Skills: Machine Learning, Python, TensorFlow, PyTorch Salary: Not disclosed