Required Senior Data & ML Engineer - 3D Perception Team
About the team:
Join PUMA, the team developing the deep-learning perceptions solutions at the heart of our 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.
What will your job look like?
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.
Requirements: All you need is:
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.
Familiarity with PyTorch and ML workflows, including datasets, training, checkpoints, evaluation, and GPU computation.
Strong ownership, independent problem-solving, and cross-team collaboration skills.
Nice to have:
Master's degree in a relevant field.
Experience in computer vision or autonomous-driving systems.
Familiarity with model deployment, hardware-aware ML, or specialized AI accelerators.
This position is open to all candidates.