G

Gatik AI

Autonomous trucking for middle-mile freight

Senior Software Engineer - ML Infrastructure

Applied Machine LearningPosted on Nov 13
Full-timeSanta Clara, CA, USA$180k - $260kIn-person

About the Job

Requirements

  • A bachelor's degree in Computer Science, Computer Engineering, Electrical Engineering, Robotics, or a related field and 5 or more years of relevant industry experience, or a master's degree and 3 or more years of relevant industry experience. Staff candidates typically have 8 or more years of experience, including time setting technical direction beyond a single team.
  • Strong Python software engineering skills, including designing modular, maintainable, high-performance, well-tested systems in a shared codebase.
  • Experience building end-to-end machine learning systems, from data pipelines and dataset curation through training, evaluation, and production, using tools such as SQL, pandas, PyTorch, or scikit-learn.
  • Hands-on production experience with Ray, including Ray Core, Ray Data, Ray Train, and/or Ray Tune, and running and debugging Ray workloads at scale.
  • Deep experience with PyTorch and PyTorch Lightning for large-scale training, including distributed strategies such as DDP or FSDP, mixed precision, and checkpointing.
  • Experience designing and operating scalable machine learning infrastructure, such as distributed training and evaluation pipelines, experiment tracking, model serving, or large-scale distributed jobs.
  • Familiarity with shared-compute environments, such as GPU scheduling and quota, job orchestration, and multi-tenant clusters.
  • Solid grounding in statistics and experimental design, including choosing evaluation strategies, metrics, and thresholds that support real-world decisions.
  • Clear communication and the ability to work effectively across software, machine learning, and autonomy teams.

Responsibilities

  • Design and maintain reproducible pipelines that discover, label, and organize driving scenarios from the fleet and simulation, with versioning, provenance, well-designed train, validation, and holdout splits, and clear data contracts.
  • Build balanced training and evaluation sets covering truck scenarios across routes, weather, and traffic conditions, and analyze coverage, composition, and label quality to shape dataset and model priorities.
  • Build distributed training, inference, and evaluation pipelines with robust data loading, checkpointing, and experiment tracking that run on multi-tenant GPU clusters.
  • Implement model artifact management, experiment tracking, and automated model testing and deployment practices for adoption across teams.
  • Define, implement, and validate model and release metrics, including precision, recall, F1, ROC AUC, calibration, and task-specific metrics, and characterize behavior across software releases and scenario types.
  • Set up baselines, ablations, and holdout comparisons; quantify uncertainty; and communicate results that help the autonomy team make release decisions.
  • Develop dashboards and services for comparing software versions, exploring scenario-level results, visualizing model outputs, and gathering reviewer feedback.
  • Partner with ML Platform, Jobs Platform, Cloud Platform, and high-performance computing engineers to design and review changes to GPU scheduling and quota, job orchestration, storage, and compute cost.
  • Design clean interfaces between data, modeling, and product layers; champion testing, type checking, code review, and fast continuous integration; and lead incident root-cause analysis and prevention, including cross-team incidents.
  • At the Staff level, define and drive multi-quarter technical strategy for data and training infrastructure, represent the team in architecture discussions, and mentor engineers across the team.

About the Company

Gatik AI

Autonomous trucking for middle-mile freight

Gatik AI provides autonomous trucking services for middle-mile B2B freight. Its Level 4 driverless trucks move goods on fixed, short-haul routes between distribution centers and retail or foodservice locations, offered as Autonomous Transportation-as-a-Service that integrates into customers' supply chains. Operations span Texas, Arizona, Arkansas, and Canada with clients such as PepsiCo and Loblaw. The company reports over $600 million in contracted revenue and about 85,000 fully driverless orders completed, aiming to grow to more than 100 driverless trucks by end of 2026. Gatik’s approach focuses on repeatable routes and a service-based model to scale its fleet, technology, and infrastructure through commercial partnerships.

Company size
201-500
Company stage
Series D
Total funding
$351.7M
Headquarters
Santa Clara, CA
Founded
2017

Recently Posted Jobs

Explore more opportunities at Gatik AI

See more jobs