Machine Learning Engineer

The role

You will develop, optimize, and deploy production ML models across the autonomous vehicle stack. This includes owning the full ML lifecycle from data strategy to real-time deployment on vehicle hardware.

What you will do

• Own the full ML lifecycle including data strategy, training, and deployment
• Develop and improve models for perception, prediction, planning, and scene understanding
• Optimize models using quantization, pruning, sparsification, and compression
• Integrate trained models into C++-based autonomy systems
• Profile and optimize neural networks using CUDA and TensorRT
• Analyze model performance using simulation and real-world driving data
• Build high-throughput pipelines for training, evaluation, and data processing
• Develop pipelines for dataset curation, annotation, and continuous feedback
• Partner with autonomy, systems, hardware, and infrastructure teams

Required qualifications

• MS or PhD in Computer Science, Machine Learning, Robotics, Electrical Engineering, Statistics, Optimization, or a related field
• 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
• Deep understanding of ML workflows including data curation, training, evaluation, and deployment
• Experience deploying and optimizing neural networks for real-time, embedded, robotics, autonomous driving, or other performance-constrained systems
• Experience with model optimization techniques such as quantization, pruning, and compression
• Experience with software architecture, profiling, latency optimization, system-level debugging, and data flow analysis

Nice to have

• Experience with CUDA and TensorRT
• Experience with cloud-based ML training and evaluation pipelines, preferably Azure
• Experience with transformers, multimodal models, diffusion models, world models, or end-to-end driving models
• Experience in autonomous driving, robotics, or other safety-critical real-time ML systems
• Publications or demonstrated technical contributions in efficient ML, autonomous driving, robotics, or related areas
• Prior contributions to large-scale ML systems deployed in production

About the company

Gatik is a leader in autonomous middle-mile logistics providing autonomous transportation-as-a-service. The company uses proprietary Level 4 autonomous technology to transport freight for retailers.