Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

AI/ML Compiler Engineer, High-Performance GPUs

Advanced Micro Devices , Inc.

Advanced Micro Devices in San Jose, CA is seeking a highly motivated PMTS AI/ML Compiler Engineer to drive compiler and performance optimizations for AI training and inference on AMD GPU platforms. You will work at the intersection of compiler tech, ML frameworks, GPU architecture, and performance engineering within the ROCm and ONNX Runtime stacks. The ideal candidate combines deep compiler expertise with hands-on AI workload optimization, collaborates with architecture, kernel, runtime, and #J-18808-Ljbffr

Vacancy posted 1 day ago
Similar jobs that could be interesting for youBased on the AI/ML Compiler Engineer, High-Performance GPUs in San Jose, CA vacancy
  •  ...Semiconductor's AGI Computing Lab in San Jose seeks an engineer to design and implement ML compilers for high-performance DL workloads, collaborating with hardware teams to...  .... The role emphasizes collaboration, research, and practical impact for AI workloads. #J-18808-Ljbffr
    Performance

    Conductor

    San Jose, CA
    1 day ago
  • $165.2k - $223.6k

     ...want to be part of AI revolution? At AWS our...  ...that optimizes the performance of complex ML models executed on AWS...  ...is for a software engineer in the Compiler team for AWS Neuron....  ...making Neuron compiler highly performant and easy-...  ...accelerators (e.g., GPUs) is preferred but... 
    Performance
    Internship
    Local area
    Flexible hours

    Amazon

    Cupertino, CA
    4 days ago
  • $165.2k - $223.6k

    The AWS Neuron Compiler team is actively seeking skilled compiler engineers to join our efforts in...  ...for advanced ML capabilities, powering...  ...solutions like Generative AI. In this role as a...  ...system-wide performance optimization. As...  ...You’ll also:Build high-impact solutions to... 
    Performance
    Internship
    Local area
    Flexible hours

    Amazon

    Cupertino, CA
    4 days ago
  • $165.2k - $223.6k

     ...ML Compiler EngineerAWS Machine Learning accelerators are...  ...-in-class ML training performance with the most...  ...resources.As an ML Compiler Engineer with the Annapurna Neuron...  ...to market, and ship high-quality code.A background...  ...in Machine Learning, AI accelerators, or... 
    Performance
    Internship
    Local area
    Relocation
    Flexible hours

    Amazon

    Cupertino, CA
    3 days ago
  • $184.7k - $324.8k

     ...On-Device ML Compiler Engineer, Model Compilation, Graphics, Games and Machine...  ...States Machine Learning and AI Imagine being at the...  ...model authoring, runtime, and performance teams to ensure that models...  ...use cases. We’re seeking a highly motivated software engineer... 
    Performance
    Relocation

    Apple Inc.

    Cupertino, CA
    5 hours ago
  • $152k - $241.5k

     ...Learning Applications and Compiler Engineer!NVIDIA is seeking...  ...develop, and maintain high-performance runtime and compiler components...  ...at top tier ML, compiler, and computer...  ...architectures, such as GPUs, spatial accelerators,...  ...Experience with large-scale AI distributed inference... 
    Performance
    Full time

    Nvidia

    Santa Clara, CA
    4 days ago
  • $193.3k - $261.5k

     ...forefront of maximizing performance for AWS's custom ML accelerators....  ...software boundary, our engineers craft high-performance kernels...  ...what's possible in AI acceleration.The AWS...  ...toolkit includes an ML compiler, runtime, and...  ...for ML or HPC such as GPUs, CPUs, FPGAs, or custom... 
    Performance
    Internship
    Local area
    Work from home
    Flexible hours

    Amazon

    Cupertino, CA
    5 days ago
  • $156k - $387.6k

     ...Machine Learning Engineer - AI Compiler Optimization Location: San Jose Team: Technology Employment...  ..., with the ability to independently perform performance bottleneck analysis and...  ...handle compilation optimization needs in high-concurrency scenarios. Experience... 
    Performance
    Temporary work
    Local area

    ByteDance

    San Jose, CA
    2 days ago
  • $207k - $300k

    Deliver certified performance benchmarks six months...  ...-pilot goals under compiler constraints.Partner...  ..., and Cloud AI Systems to steer multi...  ...technical proficiency ML workload performance...  ..., familiarity with high level compiler...  ....Google's software engineers develop the next-generation... 
    Performance
    Worldwide

    Google

    Mountain View, CA
    4 days ago
  • $150.4k - $277.6k

     ...Machine Learning Compiler EngineerAt Apple, we're on...  ...pushing the boundaries of AI and hardware...  ...Machine Learning Compiler Engineer on the Apple Neural Engine...  ...you'll work to bring high-performance, low-power AI solutions...  ...specialized SoCs or GPUs, and knowledge of deep... 
    Performance
    Relocation

    Apple

    Sunnyvale, CA
    1 day ago
  •  ...Annapurna Labs seeks a Sr. ML Kernel Performance Engineer to advance AWS Neuron on Inferentia...  ...Trainium. You will design high-performance kernels and...  ..., collaborating with compiler, runtime, framework, and hardware...  ...exposure to cutting-edge AI acceleration and... 
    Performance

    Amazon

    Cupertino, CA
    5 days ago
  •  ..., is seeking extraordinary machine learning and GPU programming engineers to deliver robust compute solutions on Apple Silicon. You will contributes to optimized kernels, ML workflows, and high-performance GPU architectures across iOS, macOS and Apple TV. The role emphasizes... 
    Performance

    Apple

    Cupertino, CA
    1 day ago
  •  ...Apple is seeking an ML Infrastructure Engineer to advance on-device ML by building graph compilers and runtimes for Apple devices. You will shape the infrastructure enabling...  ..., and execution on Apple Silicon, pushing performance and efficiency in constrained environments.... 
    Performance

    Apple

    Cupertino, CA
    4 days ago
  • $160.5k - $240.7k

     ...Technologies, Inc. Job Area Engineering Group, Engineering...  ...Summary Qualcomm AI Hub is the platform for...  ..., optimize, and deploy ML models on Qualcomm...  ...Workbench lets developers compile trained PyTorch or ONNX...  ...regressions, and performance degradations; develop clear... 
    Performance

    Qualcomm

    Santa Clara, CA
    5 hours ago
  • $100k

     ...industry on cutting-edge AI technology, revolutionizing performance expectations, ease...  ...in software models, compilers, platforms,...  ...technologists have developed a high performance RISC-V...  ...an Physical Design Engineer to lead cross-...  ...integrate, and deploy AI/ML-driven solutions... 
    Performance
    Permanent employment

    Tenstorrent

    Santa Clara, CA
    1 day ago
  •  ...CA is seeking an Open Source Software Engineer to build and optimize ML systems that run on AMD hardware. You will bridge open source AI software with GPU systems, design, implement...  ...ML frameworks. The role emphasizes performance, kernel fusion, and collaboration in open... 
    Performance

    Advanced Micro Devices , Inc.

    San Jose, CA
    1 day ago
  •  ...world's largest AI chip, 56 times larger than GPUs. This architecture...  ...with ultra high-speed inference....  ...The RoleThe Core ML team develops novel...  ...Wafer-Scale Engine. Our work spans...  ...ML frameworks, compilers, runtimes, and low...  ...capabilities, diagnose performance bottlenecks, and... 
    Performance

    Cerebras Systems

    Sunnyvale, CA
    3 days ago
  • $193.3k - $261.5k

     ...Senior Software Development Engineer to own the design and...  ...hardware, developing high-performance compute kernels,...  ...compute kernels for a custom ML accelerator architecture...  ...compute kernels for GPUs, DSPs, or custom accelerators...  ...early adopter of AI-assisted development tools... 
    Performance
    Local area
    Flexible hours

    Amazon

    Cupertino, CA
    3 days ago
  •  ...scientific discovery to powering AI and the technologies people...  ...has a broad portfolio of several ML accelerator devices with NPUs (...  ...based of AMD's proprietary AI-engine architecture.THE PERSON:The ideal...  ...issues in a complex compiler tool stack. This role will be in... 

    AMD

    San Jose, CA
    4 days ago
  •  ...is seeking an experienced software engineer to design, develop, and optimize ML models for AWS Neuron on Inferentia...  ...frameworks and kernels, collaborating with compiler, runtime, and customer teams to maximize performance. You will lead high-performance kernel development,... 
    Performance

    Amazon

    Cupertino, CA
    4 days ago
  •  ...scientific discovery to powering AI and the technologies...  ...Team is seeking an AI/ML Solutions Engineer to empower customers and...  ...integration, deployment, and performance analysis, ensuring...  ...scalable AI solutions on AMD GPUs.THE PERSON: A highly motivated and passionate... 
    Performance

    AMD

    Santa Clara, CA
    5 days ago
  • $184.7k - $324.8k

     ...On-Device ML Quality Infrastructure Engineer, Graphics, Games & ML Cupertino...  ...Learning and AI The On-Device Machine...  ..., machine learning compilers and runtimes to execute...  ..., and ensuring high-quality, low-friction...  ...fuzz testing), running performance benchmarks, and... 
    Performance
    Relocation

    Apple Inc.

    Cupertino, CA
    3 days ago
  • $224k - $356.5k

     ...a Machine Learning Engineer to join the GPU accelerated...  ...for ETL, SQL, and ML/DL model training...  ...use cases. NVIDIA GPUs offer a promising...  ...the latest ML/AI methods to empower...  ...learning solutions for performance prediction and optimization...  ...productionizing high-quality ML/DL... 
    Performance
    Full time

    Nvidia

    Santa Clara, CA
    5 days ago
  •  ...ML Systems Research EngineerAdvance your...  ...to powering AI and the technologies...  ...Systems Research Engineers to build the reinforcement...  ..., measuring performance, managing long-latency...  ...long-horizon and high-latency reward...  ...that interact with compilers, profilers, simulators... 
    Performance

    Advanced Micro Devices , Inc.

    Santa Clara, CA
    3 days ago
  • $173k - $253k

    Matterport - Senior ML Ops Engineer Job Description CoStar...  ...in enhancing the performance, efficiency, and scalability...  ..., and deploying highly efficient models into...  ...hardware platforms (e.g., GPUs, CPUs, edge devices)....  ..., and efficient AI.Contribute to the continuous... 
    Performance
    Full time
    Work at office
    Work from home

    Matterport

    Sunnyvale, CA
    3 days ago
  • $150.4k - $277.6k

     ...Graphics and Displays org provides a suite of high-performance data parallel algorithms for developers...  ...machine learning and GPU programming engineers who are passionate about providing...  ...Shaders. Performing in-depth analysis, compiler and kernel level optimizations to... 
    Performance
    Relocation

    Apple Inc.

    Cupertino, CA
    5 hours ago
  •  ...Qualcomm Technologies, Inc. is seeking a Machine Learning Engineer to create and optimize ML compiler technologies that translate PyTorch/ONNX models into efficient code for CPU, GPU, and NPU devices. You will work with cross-functional teams to advance ML frameworks... 

    Qualcomm

    Santa Clara, CA
    1 day ago
  •  ...computing experiences—from AI and data centers, to...  ...:We are hiring AI / ML Platform Engineers to build the platform...  ..., observability, performance, and developer experience...  ...integrations with compilers, ROCm/HIP tooling,...  ...shared GPU clusters or high-performance ML infrastructure... 
    Performance

    AMD

    Santa Clara, CA
    5 days ago
  • Company DescriptionAxiado is an AI-enhanced security processor...  ...the roleWe're looking for an ML engineer who works across the full...  ...optimizing training and inference performance on GPU/AI-accelerator...  ...persistence, intelligence and high curiosity over pedigree alone... 
    Performance

    Axiado Corporation

    San Jose, CA
    5 days ago
  • $174.72k - $295.68k

     ..., integrating advanced AI and autonomous driving...  ...seeking Machine Learning Engineers with strong expertise...  ...robotics data.Develop high-quality multi-view future...  ...Language-Action (VLA) driving performance.Extend prediction...  ...mixed precision, torch.compile, kernel-level... 
    Performance
    Full time

    XPENG Motors

    Santa Clara, CA
    5 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to AI/ML Compiler Engineer, High-Performance GPUs. Be the first to apply!