AI/ML Compiler Engineer, High-Performance GPUs
Advanced Micro Devices
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-LjbffrVacancy 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
- Senior / Staff ML Compiler Engineer Location: San Jose or Irvine... ...processors designed for edge AI, wireless... ...opportunity to join a highly innovative engineering... ...technologies that unlock the performance of next-generation... ...accelerators, NPUs, DSPs, GPUs, or custom compute...PerformanceFull timeSan Jose, CA25 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...PerformanceInternshipLocal areaFlexible hours$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...PerformanceInternshipLocal areaFlexible hours$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...PerformanceFull time$213k - $263k
...architect and create high-performance custom silicon; we develop... ...report to a Software Engineering Manager. You will... ...Leverage agentic AI to accelerate analysis... ...our production grade compiler, and design and implement... ...experience working with ML inference or linear...PerformanceFull timeRemote work$193.3k - $261.5k
...building Generative AI on AWS. The Inferentia... ...delivers best-in-class ML inference performance at the lowest cost in... ...which includes an ML compiler, runtime and natively... ...including silicon engineering, hardware design and... ...BalanceOur team puts a high value on work-life balance...PerformanceInternshipLocal areaWork from homeRelocationFlexible hours- ...focuses on optimizing the computational performance of recommendation/inference systems, enabling powerful ML compute for internal businesses... ...Successful candidates will design high-performance kernels, drive model compilation optimization, and collaborate with hardware...Performance
Bytedance
San Jose, CA16 hours ago $165.2k - $223.6k
...forefront of maximizing performance for AWS's custom ML accelerators. Working at... ...hardware-software boundary, our engineers craft high-performance kernels for... ...of what's possible in AI acceleration.The AWS Neuron... ...toolkit includes an ML compiler, runtime, and application...PerformanceInternshipLocal areaWork from homeFlexible hours$124k - $195.5k
...Learning Software Engineer, TensorRT PerformanceNVIDIA... ...and improving the performance of NVIDIA's... ...are using NVIDIA GPUs to power a... ...areas like Generative AI, Recommenders and... .... Implement graph compiler algorithms, frontend... ...opportunity employer. As we highly value diversity in...PerformanceNVIDIA
Santa Clara, CA2 days 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...PerformancePermanent employment- ..., 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
$193.3k - $261.5k
...Overview Do you want to be part of the AI revolution? At AWS we create... ...Neuron is an SDK that optimizes the performance of complex ML models on AWS Inferentia and Trainium... ...This role is for a senior software engineer in the Compiler team responsible for building the next...PerformanceFlexible hoursAmazon
Cupertino, CA16 hours ago$165.2k - $223.6k
...a Software Development Engineer to own the design and implementation... ...hardware, developing high-performance compute kernels,... ...kernels for a custom ML accelerator... ...developing compute kernels for GPUs, DSPs, or custom accelerators... ..., Neuron, TPU or other AI acceleration hardware-...PerformanceLocal areaFlexible hours$86k - $148k
...Role: We're looking for AI/ML Software Engineer to lead workload-driven... ...as a technical authority on performance and architecture tradeoffs.... ...interconnect) and software teams (compilers, runtimes, ML frameworks).... ...and retention of highly qualified people. Artificial...PerformanceFull timeWork at officeLocal area- ...evolution where innovative AI meets the elegance of... ...of research, software engineering, hardware engineering,... ..., building ML compilers and runtimes for efficient... ...with strong, competitive performance on Apple platforms. The... ...subject area. Highly proficient in Python programming...Performance
Socket.dev
Cupertino, CA16 hours ago - ...computing experiences—from AI and data centers, to... ...are hiring AI / ML Platform Engineers to build the platform... ...reproducibility, observability, performance, and developer... ...integrations with compilers, ROCm/HIP tooling,... ...shared GPU clusters or high-performance ML infrastructure...Performance
- ...computing experiences—from AI and data centers, to... ...Machine Learning Engineer to join our Models and... ...models on a large number of GPUs, and if you are passionate... ...-end training pipeline performance.Optimize the... ...EXPERIENCE:Experience with ML/DL frameworks such as PyTorch...Performance
$272k - $431.25k
...for a Machine Learning (ML) Engineer to join the GPU... ...and use cases. NVIDIA GPUs offer a promising avenue... ...will apply the latest ML/AI methods to empower enterprises... ...learning solutions for performance prediction and... ...implementing, and productionizing high-quality ML/DL solutions...PerformanceFull time$174.72k - $295.68k
..., integrating advanced AI and autonomous driving... ...seeking Machine Learning Engineers with strong expertise... ...robotics data. Develop high-quality multi-view... ...Language-Action (VLA) driving performance. Extend... ...mixed precision, torch.compile, kernel-level optimization...PerformanceFull time$184k - $356.5k
...seeking a Senior MLOps Engineer to join our DSX... ...groundbreaking AI workloads. We partner... ...on the demands of ML workloads, help... ...initiatives, including LLM performance evaluation and... ...hypervisor or OS, compilers or runtimes,... ...opportunity employer. As we highly value diversity in...PerformanceFull time$155.42k - $205.9k
...DescriptionAbout the Team: The AI Validation Platform... ..., with a focus on performance, availability,... ...development by prioritizing high-impact, ML-centric use cases. About... ...ML Infrastructure engineer to help build and scale... ...utilization of cutting-edge GPUs, while also leveling...PerformanceFull timeLocal areaRemote workWork from homeRelocationRelocation packageFlexible hours$153.2k - $234.1k
...Join the Embodied AI Infra Foundation team... ...machine learning engineer working on our cutting... ...the safety and performance of the car, rather... ...vehicles.As a Senior ML Infra Engineer, you... ...interviewing with high, well-calibrated standards... ...large clusters of GPUs/CPUs or other...PerformanceFull timeWork at officeLocal areaRemote workWork from homeRelocationRelocation packageFlexible hours$55 - $60 per hour
...with deep proficiency in AI-centric libraries such... ...to enhance model performance using external knowledge... ...management, dynamic prompt engineering, and secure data handling... ...data handling. Skills: ML. Python. TensorFlow.... ...staffing, powered by a high-performing recruitment...PerformanceTemporary workLocal areaCynet Systems
Santa Clara, CA17 hours ago- ...Neuron team in Cupertino, CA seeks a senior software engineer to advance the next generation of a compiler transforming PyTorch, TensorFlow and JAX models into... ...optimization passes, collaborate across chip, runtime, and ML teams, and contribute to open‑source communities...
Amazon
Cupertino, CA16 hours ago - ...Description: We are looking for a senior ML infrastructure engineer to build and evolve the systems that... ...of heterogeneous computing and high-performance computing, especially for ML... ...Docker, CI/CD, cloud infrastructure, GPUs, and internal platform tooling Bonus...Performance
Maxinsights
Santa Clara, CA28 days ago $150k - $200k
...Management," embodies our commitment to high ethical standards and transparent... ...challenges with strong emphasis placed on performance and skill—and equal, merit-based opportunities... ...roleWe are seeking a backend-focused AI/ML engineer to support fast-moving proof-of-concept...PerformanceFull timeContract workTemporary workFor contractorsLocal areaImmediate start$2,000 per month
...CAEtched is building AI chips that are... ...be impossible with GPUs, like real-time video... ...into maximally performant instruction sequences... ...Fit If You HaveAn ML Research... ...kernels, the CUDA compilation stack and related... ...and greatly value engineering skills. We do not...PerformanceETCHED LLC
Cupertino, CA2 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 ensure...PerformanceRelocation$150k - $350k
...verification with agentic AI workflows. Our platform... ...generative AI to assist engineers in RTL design, simulation... ...Overview We are seeking an ML Systems Engineer to optimize the performance and efficiency of large language... ..., GPU computing, or high‑performance inference...PerformanceChipAgents
San Jose, CA3 days ago$116k - $224.25k
...are searching for a Backend Compiler Engineer for an exciting and fun role... ...constructing and emitting the highest performance GPU machine instructions for... ...learning ignited modern AI — the next era of computing... ...opportunity employer. As we highly value diversity in our...PerformanceWorldwide
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!
Related searches
- ai developer San Jose, CA
- ai engineer San Jose, CA
- ai prompt engineer San Jose, CA
- senior ai engineer San Jose, CA
- ai engineer remote San Jose, CA
- senior ml engineer San Jose, CA
- computer vision machine learning engineer San Jose, CA
- machine learning engineer San Jose, CA
- machine learning software engineer San Jose, CA
- performance test architect San Jose, CA