RESEARCH SCIENTIST - AGENTIC RL - REMOTE

JDA WMS PTE. LTD.; Singapore; Remote; SGD 90,000 - 150,000; Full time

Job summary

JDA WMS Pte Ltd in Singapore is seeking a specialist in Reinforcement Learning to design reward mechanisms and align short-term agent decisions with long-term outcomes that impact customer satisfaction and retention.

Collaborate with infrastructure and engineering teams to turn research ideas into production-tested agent behaviors, and perform offline evaluation and ablation studies using PyTorch or JAX. Experience with RL benchmarks and solid Python coding are essential.

Qualifications

• Reinforcement Learning Expertise: Strong research experience in RL, applying techniques to real-world problems.
• Programming & ML Skills: Proficiency in Python and modern ML/RL frameworks such as PyTorch, JAX, or similar tools.
• Strong Analytical & Research Skills: Ability to critically evaluate metrics, investigate benchmark results, and conduct rigorous experiments.

Responsibilities

• Reward Design & Long-Term Decision Making: Develop reward mechanisms and connect short-term decisions with delayed outcomes like satisfaction and retention.
• User Modeling & Adaptive Agent Behavior: Design systems that understand changing user preferences and uncertainties, determine responses.
• Policy Optimization & Evaluation: Develop offline evaluation methods, strong baselines, and ablation studies to improve agent performance.
• Research & Engineering Collaboration: Work with infrastructure and engineering teams to translate research into production agent behaviors.

Skills

Reinforcement Learning Expertise

Strong Analytical & Research Skills

Tools

Python

PyTorch

JAX

Job description

JOB RESPONSIBILITIES

• Reward Design & Long-Term Decision Making: Develop effective reward mechanisms and solve long-term credit assignment challenges by connecting an agent’s short-term decisions with delayed outcomes, such as customer satisfaction, product returns, and user retention.

• User Modeling & Adaptive Agent Behavior: Design systems that can understand changing user preferences and behaviors, manage uncertainty, and determine the appropriate response, including asking questions, making recommendations, taking actions, or waiting for more information.

• Policy Optimization & Evaluation: Develop reliable offline evaluation methods, establish strong experimental baselines, and conduct ablation studies to measure and improve agent performance. Explore advanced RL techniques as more real-world data becomes available.

• Research & Engineering Collaboration: Work closely with infrastructure and engineering teams to transform complex product challenges into research problems and implement research ideas as testable agent behaviors in real-world production environments.

REQUIREMENTS

• Reinforcement Learning Expertise: Strong research experience in Reinforcement Learning, with a track record of applying RL techniques to practical, real-world problems.

• Programming & ML Skills: Strong proficiency in Python and hands-on experience with modern ML/RL frameworks such as PyTorch, JAX, or similar tools.

• Strong Analytical & Research Skills: Ability to critically evaluate metrics, investigate benchmark results, identify potential limitations, and conduct rigorous experiments and ablation studies.

JDA WMS Pte Ltd | EA Personnel: Pham Thi Tuyet Mai

EA License No: 23S1595 | EA Registration No: R25127838