Software Engineer, AI Libraries
Core
Build modular AI libraries and platforms to support training, evaluation, and deployment of autonomous driving models at scale on large GPU clusters.
Role type
Senior IC software engineer (AI platform infrastructure)
Builds
Reusable software libraries, tools, and abstractions for distributed training and inference
Domain
Autonomous driving / AI infrastructure
Deliverable
production ML models
Required skills
Python, software architecture, testing, observability, concurrent/parallel/distributed computing, ML frameworks (PyTorch/TensorFlow), cloud environments (Azure)
Preferred skills
Experience with multimodal data, camera/radar/lidar data sources, large-scale GPU clusters
Technologies
Python, PyTorch, TensorFlow, Azure (via careerplan.io/jobs/82f389e1-b856-4157-ab24-f83f467329e5-software-engineer-ai-libraries-at-wayve)
Responsibilities
Design and write Python code for scalable AI platforms, improve engineering standards, support ML teams in scaling model performance, build modular libraries for data loading and distributed training
Seniority
Senior, hands-on IC
