Machine Learning Engineer - Remote

YO AI Labs is seeking a Senior Software Engineer to support an AI training project by creating reinforcement learning environments for evaluating AI models on complex software engineering tasks. The role involves designing reproducible scenarios, deterministic verification systems, and reference solutions, and assessing AI agents’ reasoning and use of MCP tools.

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 use MCP tools effectively
• Develop realistic, reproducible software engineering scenarios
• Ensure tasks accurately measure coding ability, problem-solving, and tool usage
• Document solutions and provide clear technical feedback

Requirements:

• Strong proficiency in Python 3, Java, Rust, C++, or TypeScript
• Strong understanding of algorithms and data structures
• Experience with bug fixing and debugging complex software issues
• Proven experience in feature implementation and codebase refactoring
• Strong knowledge of performance optimization and tuning
• Excellent written and verbal communication
• Strong attention to detail
• Experience working with large-scale or distributed codebases
• Familiarity with AI/ML systems is a plus but not required
• Experience with rigorous code reviews and software engineering best practices
• Experience working effectively in remote or cross-functional teams