Senior/Staff ML Infrastructure Engineer

Key Skills

Algorithms, Azure, C++, CUDA kernels, Infrastructure as code, KPI, Machine learning techniques, Optimization, Python Programming, PyTorch

Job Description

About the role:

We're looking for high-energy, creative, and collaborative candidates who want to work in a fast-paced, execution-oriented team. You will play an essential role in helping accelerate the development and deployment of our AV software stack. This position has a strong technical background, hands-on software engineering experience, and a knack for solving hard problems.

This role is onsite 4 days a week at our Mountain View, CA office!

What you'll do:

• Own development of ML models end-to-end from data strategy, initial development, optimization, production platform validation, and fine-tuning based on metrics and on-road performance
• Lead efficient neural network development including quantization, pruning, sparsification, compression, and novel differentiable compute primitives
• Build the foundation models for the on-vehicle and offline applications; Develop metrics and tools to analyze errors and understand improvements in our systems
• Train and evaluate DNNs for the purpose of benchmarking neural network optimization algorithms – optimizing for latency and power consumption
• Design and implement a horizontally scalable, high-throughput cloud inference pipeline for evaluation and KPI calculation
• Streamline workflows to allow creation of verified, deployable artifacts from annotated data
• Support data preparation for training: building a horizontally scalable data preparation pipeline that is simple to use and doesn't delay training
• Support development of tools for introspection and visualization to understand what is going well and what can be improved in our work

What we're looking for:

• Bachelor's Degree in Computer Science, Machine Learning or relevant field
• Master's Degree with a focus on Machine Learning, Statistics, Optimization or a related field (preferred) or relevant work experience
• 7+ years of experience working with large ML projects and/or building production ML systems
• Excellent C++, Python, and/or CUDA programming skills
• Familiarity with modern machine learning environments such as Pytorch
• Expert experience with optimization techniques from high-level ML algorithms to low-level HW utilization
• Experience in software architecture, system performance, latency, and data flow
• Expert experience in machine learning workflows: data sampling and curation, pre-processing, model training, ablation studies, evaluation, deployment, inference optimization
• Strong analytical skills, especially for performance troubleshooting (e.g. profiling, roofline model)
• Industry experience in building large-scale ML pipelines
• Experience with cloud ML training pipelines in Azure (preferred)
• High Performance Computing experience (preferred)

More about Gatik:

With headquarters in Mountain View, CA and offices in Canada, Texas and Arkansas. Gatik is establishing new standards of success for the autonomous trucking industry every day. Visit us at Gatik for more company information and Jobs @ Gatik for more open roles.

Gatik News:

• The 10 most innovative companies in transportation of 2023
• America’s Best Startup Employers of 2023 by Forbes
• America’s Best Startup Employers of 2022 by Forbes
• Gatik been named as a 2023 FreightTech 25 winner!
• Gatik named a TIME Best Invention of 2022
• Apeksha Kumavat recognized on the Inc. 2022 Female Founders 100 List
• Gatik’s Gautam Narang on the importance of knowing your customer
• Gatik CEO Gautam Narang was Named Among the Most Exceptional Entrepreneurs of 2022 by Goldman Sachs

Taking care of our team:

At Gatik, we connect people of extraordinary talent and experience to an opportunity to create a more resilient supply chain and contribute to our environment’s sustainability. We are diverse in our backgrounds and perspectives yet united by a bold vision and shared commitment to our values. Our culture emphasizes the importance of collaboration, respect and agility.