Product Engineer, AI Data Platform

Shape the Future of AI

At Labelbox, we're building the critical infrastructure that powers breakthrough AI models at leading research labs and enterprises. Since 2018, we've been pioneering data-centric approaches that are fundamental to AI development, and our work becomes even more essential as AI capabilities expand exponentially.

About Labelbox

We're the only company offering three integrated solutions for frontier AI development:

1. Enterprise Platform & Tools: Advanced annotation tools, workflow automation, and quality control systems that enable teams to produce high-quality training data at scale
2. Frontier Data Labeling Service: Specialized data labeling through Alignerr, leveraging subject matter experts for next-generation AI models
3. Expert Marketplace: Connecting AI teams with highly skilled annotators and domain experts for flexible scaling

Why Join Us

• High-Impact Environment: We operate like an early-stage startup, focusing on impact over process. You'll take on expanded responsibilities quickly, with career growth directly tied to your contributions.
• Technical Excellence: Work at the cutting edge of AI development, collaborating with industry leaders and shaping the future of artificial intelligence.
• Innovation at Speed: We celebrate those who take ownership, move fast, and deliver impact. Our environment rewards high agency and rapid execution.
• Continuous Growth: Every role requires continuous learning and evolution. You'll be surrounded by curious minds solving complex problems at the frontier of AI.
• Clear Ownership: You'll know exactly what you're responsible for and have the autonomy to execute. We empower people to drive results through clear ownership and metrics.

Role Overview

As a Product Engineer, AI Platform - Data Infrastructure at Labelbox, you will lead the design and development of our core data infrastructure, powering the seamless flow, storage, and processing of data for our AI platform. Your expertise will drive the evolution of scalable systems—anchored by high-performance databases—to support large-scale workflows, high-throughput data I/O, and streaming capabilities. You’ll enable Labelbox customers to efficiently manage and stream data for training next-generation AI models. Owning critical components of our data infrastructure, including database architecture, you’ll work end-to-end on projects from design to deployment. Collaborating cross-functionally with stakeholders, you’ll transform ideas into robust, scalable solutions that enhance platform adoption and customer success.

Your Impact

• Design and build scalable data infrastructure, integrating high-performance databases (relational, NoSQL, cloud-native) with distributed systems for data processing, storage, and streaming.
• Optimize database systems for performance, reliability, and scalability, ensuring efficient data retrieval, indexing, and querying to support AI workflows.
• Develop and maintain data pipelines using distributed queues, message brokers, and job management mechanisms to enable high-throughput import/export operations.
• Collaborate with team members and stakeholders to align data infrastructure with platform goals and customer needs.
• Participate in Sprint Planning, Standups, and related activities to drive data-focused initiatives forward.
• Mentor and guide less experienced engineers, sharing expertise in data infrastructure and database optimization.
• Support the team’s area of ownership by working with the Support organization to resolve customer-facing data issues.
• Stay abreast of industry trends in data infrastructure and database technologies, incorporating relevant innovations into our systems.
• Contribute to technical documentation, research publications, blog posts, and presentations at conferences and forums.
• Innovation in AI: Enhance data infrastructure capabilities for an AI platform used by leading AI labs to develop powerful multi-modal large language models (LLMs).

What You Bring

• Bachelor’s degree in Computer Science, Data Engineering, or a related field. Advanced degree preferred.
• 2+ years of work experience in a software or data-focused company, with significant expertise in data infrastructure and backend engineering.
• Deep knowledge of designing and managing scalable database systems, including relational databases (e.g., PostgreSQL, MySQL), NoSQL stores (e.g., MongoDB, Cassandra), and cloud-native solutions (e.g., Google Spanner, AWS