Machine Learning Engineer

Job description

Job responsibilities

• End-to-End Model Development: Own the full ML lifecycle, including data strategy, preprocessing, training, evaluation, optimization, deployment, and monitoring.
• Autonomous Driving Models: Develop and improve models supporting perception, prediction, planning, and scene understanding.
• Efficient Neural Network Design: Optimize models using techniques such as quantization, pruning, sparsification, compression, and efficient architecture design to meet strict latency, compute, memory, and power…
• Real-Time Deployment: Integrate trained models into C++-based autonomy systems and optimize inference for production vehicle hardware.
• Model Optimization: Profile and optimize neural networks using CUDA, TensorRT, and related technologies.
• Simulation and Evaluation: Analyze model performance using simulation and real-world driving data, identify failure modes, and drive improvements.
• Scalable ML Infrastructure: Build high-throughput pipelines for training, evaluation, data processing, and large-scale offline inference.
• Data Workflows and Tooling: Develop reliable pipelines for dataset curation, annotation, preprocessing, visualization, diagnostics, benchmarking, and continuous feedback from field data.

Job requirements

• Education: MS or PhD in Computer Science, Machine Learning, Robotics, Electrical Engineering, Statistics, Optimization, or a related field.
• Experience: Open to all experience levels. Leveling will be determined based on experience and technical depth.
• Programming & Frameworks:
• Strong Python skills and experience with frameworks such as PyTorch or TensorFlow.
• Strong C++ skills and experience integrating ML models into high-performance production systems.
• Core ML & Systems Expertise:
• Deep understanding of ML workflows, including data curation, training, evaluation, ablation studies, deployment, and inference optimization.
• Experience deploying and optimizing neural networks for real-time, embedded, robotics, autonomous driving, or other performance-constrained systems.