Machine Learning Architect C2C jobs Mountain View, CA

Our customer is a software development team with the top GERMAN OEM in automotive industry. Customer of ours is building the leading tech stack for the automotive industry and creating a unified software platform for over 10 million new vehicles per year. We're looking for talented, digital minds like you to help us create code that moves the world. Together with you, we'll build outstanding digital experiences and products for all customer brands that will transform mobility. Join us as we shape the future of the car and everyone around it.

The Chief Machine Learning Compiler Architect, within the NPU Hardware & Software organization, is intended for an individual with broad background in compiler development and architecture, with significant experience in AI/ML hardware accelerators and advanced compilation technologies. The Chief Machine Learning Compiler Architect will be responsible for designing and developing the compiler architecture for our state-of-the-art Neural Processing Unit (NPU), optimizing and transforming machine learning models into efficient executable formats that are tailored for our specialized hardware. Additionally, you will be responsible for leading research initiatives in advanced compilation techniques and driving adoption of cutting-edge optimization strategies and compilation methodologies.

Compiler Architecture & Design

Design and develop a robust compiler architecture that effectively interacts with our NPU

Implement advanced graph optimizations that incorporate both hardware agnostic and hardware specific enhancements

Develop and optimize algorithms for tiling and memory management to efficiently utilize the NPU's resources

Create sophisticated optimization passes for neural network inference and training workloads

Code Generation & Hardware Integration

Map high-level operations to optimized library macros and convert them into hardware-level instructions

Generate and manage DMA commands to facilitate data movement and operation within the hardware ecosystem

Collaborate with hardware engineers and system architects to ensure seamless integration and maximal performance of the NPU

Implement efficient scheduling and resource allocation algorithms for concurrent AI workload execution

Innovation & Technology Leadership

Stay updated with the latest trends and advancements in compiler technology and machine learning to continuously improve the compiler design

Lead research initiatives in advanced compilation techniques for AI accelerators

Drive adoption of cutting-edge optimization strategies and compilation methodologies

Mentor engineering teams on compiler design principles and best practices

Expert communicator across cultural and team boundaries

Expertise in motivating teams and fostering a collaborative and productive environment

Background in managing multiple and competing stakeholder interests; establishing trust, clear roles and responsibilities, and goodwill between partner engineering organizations

Experience managing cross-functional and/or cross-team projects

Technical leadership experience with ability to mentor engineering teams

Strategic thinking capabilities with focus on long-term architectural decisions

Collaborate and work with multiple teams across geographies and time zones

12+ years of experience in compiler development or architecture, particularly targeting AI or ML hardware accelerators

Strong understanding of machine learning algorithms and their computational implications

Working experience with TVM, IREE, XLA, MLIR or LLVM

Proficiency in programming languages such as C++ and Python

Prior experience with NPU hardware

Knowledge of automotive industry standards and functional safety requirements

Experience with neural network quantization and optimization techniques

Background in high-performance computing and parallel processing architectures

Publications or contributions to open-source compiler projects

Experience with GenAI tools for accelerated engineering workflows and AI-assisted development practices

Enthusiasm for adopting innovative AI-augmented development practices and continuous learning in rapidly evolving GenAI technologies.