Senior Data & ML Engineer - 3D Perception Team

Join PUMA, the team developing the deep-learning perceptions solutions at the heart of […]'s autonomous-driving system. Our work powers advanced, large-scale multi-task models.

We are looking for an ML Data Engineer to own the engineering foundations behind its development and training. You will build pipelines and tools that process images, video, geometric information, and evaluate model outputs. Working alongside deep-learning researchers, you will eliminate resource-intensive bottlenecks and reduce computation costs and iteration time.

• Optimize distributed data and evaluation pipelines for memory efficiency, throughput, and training performance.
• Build tools for data exploration, manipulation, validation, and quality assessment.
• Own model manipulation, packaging, distribution workflows, and interfaces with downstream deployment and EyeQ integration.
• Coordinate data contracts and workflows with researchers, infrastructure and data teams, and internal customers across […].

• Bachelor's degree in Computer Science, Software Engineering, Electrical Engineering, or a related field from a leading university.
• 4+ years of hands-on software or data-engineering experience.
• Excellent Python and Linux skills with strong software-engineering foundations.
• Experience with SQL, Spark, and PyArrow.
• Experience optimizing distributed pipelines or compute-intensive systems for memory use and parallel execution.
• Familia