AI Research Engineer (Agentic Reinforcement Learning)

About the role

As an AI Research Engineer specializing in Agentic Reinforcement Learning, you will drive research and innovation in next-generation intelligent systems. You will develop advanced reinforcement learning techniques that improve decision-making, adaptive behavior, and performance across multimodal AI architectures.

What you'll do

• Research and develop reinforcement learning algorithms for multimodal models, including diffusion-based approaches for image generation
• Design and implement scalable reinforcement learning infrastructure supporting distributed training across complex multimodal systems
• Develop and optimize reward modeling strategies to improve training stability, align model behavior with desired outcomes
• Create, curate, and maintain multimodal simulation environments and training datasets to support robust model development, evaluation, and benchmarking
• Design and implement rigorous evaluation protocols to measure model performance, monitor progress against established baselines, and validate research improvements

What you need

• Research
• Computer vision
• Deep learning
• PyTorch

Good to have

• Machine learning
• NLP