Machine Learning Engineer - Remote

YO AI Labs is seeking experienced Senior Software Engineers to support an AI training project. The role focuses on creating reinforcement learning environments, deterministic verification systems, and reference solutions for evaluating AI models on complex software engineering tasks using 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