Software Engineer
WayveRemotely
pythonazureobservabilitypytorchtensorflowgpudistributed trainingpytorch lightning
Job Description
📋 Description
- Design, build, and maintain scalable Python libraries and tools used by ML engineers and
- Develop robust abstractions for data loading, distributed training, inference, checkpointing, and
- Support training at scale across large GPU clusters and cloud-based infrastructure.
- Work closely with ML teams to understand user needs and create tools that are reliable
- Improve engineering quality across ML systems through strong software architecture, testing
- Optimise data and training pipelines to support multi-modal data sources, including camera, radar
🎯 Requirements
- Strong Python programming experience. Proven experience designing, building, and maintaining
- Strong software architecture and system design skills. Experience building tools, platforms, or
- Experience working with cloud environments, ideally Azure.
- Experience with concurrent, parallel, or distributed computing.
- Familiarity with ML frameworks such as PyTorch, TensorFlow, or PyTorch Lightning.
- Ability to work closely with technical stakeholders to refine requirements and deliver practical
🎁 Benefits
- Work on high-impact systems that directly support the development of autonomous driving technology.
- Help scale training and evaluation infrastructure across large GPU clusters.
- Build software used by ML engineers and researchers at the frontier of embodied AI.
- Join a team focused on strong engineering standards, practical abstractions, and scalable platform
- Play a meaningful role in bringing autonomous driving technology closer to real-world deployment.
Back to all jobs