Join Pokee AI as a Research Scientist and work on designing, implementing, and scaling novel reinforcement learning algorithms for their AI agent platform.

Company: Pokee AI

Role: RL AI Research Scientist

Location: Remote (US/Singapore Preferred)

Experience:

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)

Key Skills:

Reinforcement Learning
Machine Learning
Python
Deep Learning
PyTorch

Qualification:

PhD (or equivalent research experience) in Reinforcement Learning, Machine Learning, or a closely related field

Role Focus:

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

Nice to have:

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)
Experience training and fine-tuning large language models

Other:

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

RL AI Research Scientist

Skills Required

Description