Jobs · Engineering

RL AI Research Scientist

Pokee AI · United States · 1 wk ago
RemoteRemoteEngineeringFull-time

About the role

Design and implement novel RL algorithms for training AI agents on complex, multi-step enterprise workflows
Develop and refine reward modeling, context selection, and policy optimization techniques that improve agent accuracy over extended task horizons
Run large-scale experiments, analyze results rigorously, and translate research findings into production-ready components
Collaborate closely with infrastructure engineers to ensure research prototypes scale efficiently on both cloud and on-device hardware
Contribute to the company’s intellectual property through publications, patents, and open-source contributions
Stay current with the latest advances in RL, LLM fine-tuning, and AI agent architectures, and propose new research directions

Responsibilities

  • Design and implement novel RL algorithms for training AI agents on complex, multi-step enterprise workflows
  • Develop and refine reward modeling, context selection, and policy optimization techniques that improve agent accuracy over extended task horizons
  • Run large-scale experiments, analyze results rigorously, and translate research findings into production-ready components
  • Collaborate closely with infrastructure engineers to ensure research prototypes scale efficiently on both cloud and on-device hardware
  • Contribute to the company’s intellectual property through publications, patents, and open-source contributions
  • Stay current with the latest advances in RL, LLM fine-tuning, and AI agent architectures, and propose new research directions

Requirements

  • PhD (or equivalent research experience) in Reinforcement Learning, Machine Learning, or a closely related field
  • Strong publication record at top venues (NeurIPS, ICML, ICLR, AAAI, or equivalent)
  • Deep expertise in RL fundamentals: policy gradient methods, value-based methods, model-based RL, multi-agent RL, or RLHF/RLAIF
  • Proficiency in Python and at least one deep learning framework (PyTorch strongly preferred)
  • Experience training and fine-tuning large language models is a significant plus
  • Demonstrated ability to take research from prototype to production

Qualifications

  • Experience with on-device or edge inference optimization (quantization, distillation, MoE architectures)
  • Familiarity with enterprise software deployment, compliance, or regulated industries
  • Track record of open-source contributions in RL or LLM ecosystems
  • Experience with distributed training at scale (FSDP, DeepSpeed, Megatron)

Skills

  • Strong publication record at top venues (NeurIPS, ICML, ICLR, AAAI, or equivalent)
  • Deep expertise in RL fundamentals: policy gradient methods, value-based methods, model-based RL, multi-agent RL, or RLHF/RLAIF
  • Proficiency in Python and at least one deep learning framework (PyTorch strongly preferred)
  • Experience training and fine-tuning large language models is a significant plus
  • Experience with on-device or edge inference optimization (quantization, distillation, MoE architectures)
  • Familiarity with enterprise software deployment, compliance, or regulated industries
  • Track record of open-source contributions in RL or LLM ecosystems
  • Experience with distributed training at scale (FSDP, DeepSpeed, Megatron)

Benefits

  • Direct impact on the product
  • Access to cutting-edge research
  • The opportunity to shape the future of enterprise AI from the ground up

Pay

Commensurate with experience

Schedule

Full-time

Similar jobs

AI/ML/RL Scientist

Johns Hopkins Applied Physics LaboratoryLaurel, MD· 2 mo ago
Engineering$100k/yrapply on careers.jhuapl.edu