Senior ML Engineer - Remote Contract (RL Environments)

YO AI Labs; Los Angeles (CA); Remote; USD 96,000 - 165,000; Full time

Job summary

YO AI Labs is seeking a Senior Software Engineer to support an AI training project by creating reinforcement learning environments that evaluate AI models on complex software engineering tasks using Model Context Protocol (MCP) tools. You will design reproducible environments, deterministic verification, and reference solutions for tasks such as bug fixing, feature implementation, codebase refactoring, and performance optimization.

No prior AI experience is required.

Qualifications

• Proficiency in programming languages listed above.
• Experience with solving complex software problems.
• Ability to work on reinforcement learning tasks.

Responsibilities

• Create reinforcement learning environments for software engineering tasks.
• Design tasks involving bug fixing, feature development, refactoring, and performance optimization.
• Build deterministic verification systems and golden reference solutions.
• Evaluate AI agents' ability to reason through complex codebases and MCP tools.
• Develop realistic, reproducible software engineering scenarios.
• Document solutions and provide clear technical feedback.

Skills

Python

Java

Rust

C++

TypeScript

Algorithms & data structures

Bug fixing

Debugging complex issues

Feature implementation

Code refactoring

Performance optimization

Communication

Attention to detail

Job description

YO AI Labs is seeking a Senior Software Engineer to support an AI training project by creating reinforcement learning environments that evaluate AI models on complex software engineering tasks using Model Context Protocol (MCP) tools. You will design reproducible environments, deterministic verification, and reference solutions for tasks such as bug fixing, feature implementation, codebase refactoring, and performance optimization.

No prior AI experience is required.