Software Engineer - Machine Learning

What you’ll do

• Review and evaluate AI-generated machine learning code (e.g., Python, TensorFlow, PyTorch, scikit-learn) for correctness, efficiency, scalability, and clarity
• Write high-quality machine learning solutions to modeling, data processing, and deployment problems across varying difficulty levels
• Create clear, developer-friendly explanations for model architecture decisions, code logic, and problem-solving strategies
• Identify and flag edge cases or ambiguities in problem statements, datasets, or AI-generated responses

About the role

• Flexible workload — work from anywhere, on your own schedule
• High impact — your craft directly improves models used by top AI labs & Fortune 500 teams
• Clear ownership — know exactly what success looks like and have autonomy to deliver
• Growth potential — consistent high performers spearhead new programs and mentor incoming SMEs

Interview process

• Complete a screening with Zara, our AI interviewer in English, to learn more about your background and experience.
• Domain-specific Zara interview to assess your ML expertise, including ML algorithms, models, and training strategies.

About Labelbox

Labelbox builds the data engine that accelerates breakthrough AI. Our platform, expert services, and marketplace let teams iterate on data as nimbly as they iterate on code, enabling safer, smarter models in production. We’re backed by SoftBank, Andreessen Horowitz, B Capital, Gradient Ventures, Databricks Ventures, and Kleiner Perkins, and trusted by leading research labs and enterprises worldwide.