RL Environments Research Engineer

We are looking for a RL Environments Research Engineer to join our team. The ideal candidate will have experience with PyTorch or JAX, familiarity with RL concepts, and the ability to read and implement ML papers. The job involves designing and building MLE/SWE environments and tasks, and requires strong production Python skills.

Requirements

• Experience with PyTorch or JAX at the framework level
• Familiarity with RL concepts: reward functions, environment design, training loops, evaluation
• Ability to read ML papers and implement them
• Production Python skills: Docker, git, clean code, reproducible environments

Benefits

• 100% remote work
• 300 PLN to use on our benefits platform, Worksmile - gift cards, medical services, sports, etc.
• IP BOX support