Staff Software Engineer - AI Infrastructure

• Contribute to one or more of the following areas:
  • Design and optimize large-scale data processing, production and loading pipelines, supporting heterogeneous data types (images, videos, point clouds, sensor streams, etc.).
  • Build and maintain high-performance dataset management and loading frameworks, ensuring low-latency, high-throughput pipelines for training and inference.
  • Develop and optimize distributed compute and inference systems, including scheduling, resource utilization, and performance tuning.
• Collaborate with cross-functional teams (e.g. Algorithms, Data Lakehouse) to translate requirements into production-ready infrastructure solutions.
• Continuously monitor, profile, and eliminate bottlenecks across AI data, inference and compute stack.

Basic Qualifications

• Familiarity with the autonomous driving industry and enthusiasm for its challenges.
• Experience with distributed computing frameworks such as Ray, Flink or Spark.
• Experience in building and scaling ML infrastructure in cloud-native environments.
• Experience in any of the following areas:
  • Large-scale deep learning training or inference optimization focused on scalability and model acceleration.
  • Columnar storage formats (Parquet/ORC) and related ecosystems, including partitioning, compression, and vectorized I/O optimization.
  • Large-scale data loading frameworks (PyTorch Dataloader, Hugging Face Datasets).