Machine Learning Engineer - Python F/H

About the AI Studio

The AI Studio's mission is to find the fastest possible path to an autonomous supply chain. We're developing AI agents, learning systems, training models, and more to overcome the biggest challenges remaining in the global supply chain.

We're looking for an ambitious ML Engineer focused on LLMs, agents, and reinforcement learning to help build the training, evaluation, and tooling systems behind robust AI decision-making products.

You'll work across LLM fine-tuning, agent environments, reward modeling, evaluations, data pipelines, and AI workflow tooling. The role is hands-on: designing experiments, shipping production code, improving model behaviour, and building the infrastructure that lets us learn quickly from both automated and human feedback.

You'll help shape how we use LLMs inside agentic systems, how we evaluate model and agent performance, and how we turn feedback into better training data and better behaviour.

This role requires mandatory RL training experience with LLMs, including designing and iterating on rewards, reviewing LLM traces, identifying reward hacking or shortcut behaviour, and understanding when the reward signal, environment, or training process needs to change.

Design, test, and iterate on reward functions that capture the behaviors we want from LLM agents

Create data pipelines for training, fine-tuning, preference data, synthetic data generation, and human feedback collection

Develop tooling that improves how the team builds, tests, debugs, and deploys AI-assisted workflows

Experiment with RL, RLHF, RLAIF, reward shaping, policy optimization, and agent training techniques

Have experience working with AI agents and RL environments in production

Are proficient in Python and PyTorch

Hands on experience with RL techniques (reward shaping, policy optimization, RLHF)

Bonus points if:

All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability or protected veteran status.