Staff ML Compiler Engineer at General Motors | TX, USA | Rezi

Staff ML Compiler Engineer at General Motors

Staff ML Compiler Engineer

General Motors · TX, USA

1 months ago

Staff ML Compiler Engineer

General Motors · TX, USA

2 months ago
Resume preview

Impress employers and recruiters.
Choose from hundreds of resume examples.

Target Resume Now

About the Role

The AI Kernels & Compilers team is responsible for transforming cutting-edge AI research into production-grade software for autonomous vehicles. This role focuses on owning the end-to-end compilation stack, defining technical vision, and building tooling to optimize model deployment for GM's autonomous and assisted driving platforms, directly impacting the safety and efficiency of automated vehicles.

Responsibilities

  • Own and evolve the model compilation toolchain for deploying large-scale perception, prediction, and planning models to the AV.
  • Architect new compiler passes and analyses to improve build times, memory footprint, and runtime latency while preserving fidelity under strict safety and reliability constraints.
  • Collaborate with kernels, runtime, and hardware teams to co-design interfaces and shape accelerator capabilities.
  • Ensure the compiler exposes the right abstractions to unlock peak performance on each platform.
  • Set standards and best practices for model export, validation, and debugging.

Requirements

  • 5+ years of experience in compilers.
  • Experience with ML frameworks (e.g., PyTorch, TensorFlow, JAX) and software stack (e.g., ONNX, MLIR, XLA, TVM, TensorRT, etc).
  • Expertise in writing production quality Python/C++ code.
  • Expertise in the software development life-cycle - coding, debugging, optimization, testing, integration.
  • BS, or higher degree, in CS/CE/EE, or equivalent.
  • Experience building and optimizing ONNX-based model export and deployment pipelines.
  • GPU programming (CUDA) and familiarity with ML SW stack (e.g., cuDNN, cuBLAS).
  • Experience with ML accelerators and hardware architecture.
  • Experience developing and deploying machine learning models.

Skills

  • Compilers
  • ML frameworks
  • ONNX
  • MLIR
  • XLA
  • TVM
  • TensorRT
  • Python
  • C++
  • Software development life-cycle
  • CUDA
  • cuDNN
  • cuBLAS
  • ML accelerators
  • Hardware architecture
  • Machine learning models

Location

  • Remote

Work Type

  • Remote
  • Hybrid

Experience Level

  • Staff
  • 5+ years

Education Level

  • BS or higher in CS/CE/EE, or equivalent

Salary/Compensations

  • $185,100 to $335,300

Benefits

  • Medical, dental, vision
  • Health Savings Account
  • Flexible Spending Accounts
  • Retirement savings plan
  • Sickness and accident benefits
  • Life insurance
  • Paid vacation & holidays
  • Tuition assistance programs
  • Employee assistance program
  • GM vehicle discounts
  • Company vehicle evaluation program

About the Company

  • GM's vision is Zero Crashes, Zero Emissions, and Zero Congestion.
  • We are building advanced automated driving technologies, including Level 4-capable fully self-driving systems.
  • Our mission is to create safer, more sustainable, and more accessible mobility.
  • We are committed to being a workplace that fosters inclusion and belonging.
  • We believe in making the world better, safer, and more equitable for all through technology and ingenuity.

Equal Opportunity

  • General Motors is committed to being a workplace that is not only free of unlawful discrimination, but one that genuinely fosters inclusion and belonging.
  • All employment decisions are made on a non-discriminatory basis without regard to sex, race, color, national origin, citizenship status, religion, age, disability, pregnancy or maternity status, sexual orientation, gender identity, status as a veteran or protected veteran, or any other similarly protected status in accordance with federal, state and local laws.
  • General Motors offers opportunities to all job seekers including individuals with disabilities.