Systems ML Engineer - AI Compiler & Kernel Optimizations

Jobleads-US

Meta's Santa Clara-based MTIA Software Team is hiring a Software Engineer, Systems ML specializing in compilers and kernels. You will help develop the AI compiler stack, contribute to PyTorch core components, and optimize high-performance kernels for next-generation hardware accelerators.

Collaborating with AI researchers and hardware engineers, you will analyze models, implement compiler optimizations, and advance deployment velocity for training and inference on MTIA platforms.

#J-18808-Ljbffr Jobleads-US

Apply

Vacancy posted 1 day ago

Similar jobs that could be interesting for youBased on the Systems ML Engineer - AI Compiler & Kernel Optimizations in Santa Clara, CA vacancy

Sr. ML Kernel Performance Engineer, AWS Neuron, Annapurna Labs
$193.3k - $261.5k
...The Acceleration Kernel Library team is... ...AWS's custom ML accelerators. Working... ...boundary, our engineers craft high-... ...in delivering optimal performance for... ...'s possible in AI acceleration.The... ...includes an ML compiler, runtime, and... ...machine learning systems, you'll bring...
Suggested
Internship
Local area
Work from home
Flexible hours
Amazon
Cupertino, CA
2 days ago
2027 PhD ML System Engineering Intern/Co-op
...scientific discovery to powering AI and the technologies... ...highly motivated ML System Engineering co-op to join our team.... ...you to write and tune AIE kernels with IRON, so you can optimize the operators that dominate... ..., and how production AI compilers and runtimes are...
Suggested
Full time
Summer work
Internship
Summer internship
Worldwide
AMD
Santa Clara, CA
1 day ago
ML Runtime and Kernel Engineer - Core ML
Cerebras Systems builds the world's largest AI chip, 56 times larger than... ...The RoleThe Core ML team develops... ...Cerebras Wafer-Scale Engine. Our work spans... ...across ML frameworks, compilers, runtimes, and low-level kernels to implement new... ...kernel layers.Optimize computation, memory...
Suggested
Cerebras Systems
Sunnyvale, CA
4 days ago
Sr. ML Compilation Engineer
...discovery to powering AI and the... ...portfolio of several ML accelerator devices... ...AMD's proprietary AI-engine architecture.THE PERSON... ...issues in a complex compiler tool stack. This role... ...for analysis and optimization effortsContribute to... ...source control system such as GitHubKnowledge...
Suggested
AMD
San Jose, CA
4 days ago
ML Compiler Engineer, Annapurna Labs
$165.2k - $223.6k
The AWS Neuron Compiler team is actively seeking skilled compiler engineers to join our efforts in... ...is designed to optimize application models... ...innovation for advanced ML capabilities,... ...solutions like Generative AI. In this role as a... ...and deep-learning systems software....
Suggested
Internship
Local area
Flexible hours
Amazon
Cupertino, CA
1 day ago
ML Engineer, Inference & Optimization
$250k - $350k
...Staff level Inference Engineers to accelerate the... ...of Pika's AI-driven products. In... ...ensuring our creative AI systems deliver industry-... ...will design and optimize inference pipelines... ...computing kernels and distributed workloads... ...and deep learning compiler stacks.GPU & Parallelism...
Work at office
3 days per week
Pika
Palo Alto, CA
2 days ago
Software Engineer, Systems ML - Compilers / Kernels
# Software Engineer, Systems ML - Compilers / KernelsSanta Clara, CA Engineering+ 2 moreApply nowIn this... ...and part of the bigger PyTorch AI framework organization. MTIA Software... .... The compiler stack, DL graph optimizations, and kernel authoring for specific hardware, directly...
Flexible hours
Jobleads-US
Santa Clara, CA
1 day ago
On-Device ML Graph Compiler Engineer
...Apple Inc. is seeking an On-Device ML Infrastructure Engineer focused on Core ML Runtime to advance graph compilation, runtime systems, and ML operator primitives for on-device... ...on-device graph compiler, runtime, and kernels; optimize for performance, energy, and thermal...
Jobleads-US
Cupertino, CA
1 day ago
Senior ML Compiler Engineer for AI Accelerator
...seeking a motivated software engineer to join a core team... ...edge machine learning model compiler for AIE in Ryzen AI devices. The compiler translates... ...ONNX or PyTorch models into optimized code for the AIE VLIW array... .... You will work across ML frameworks, MLIR, and runtime...
Jobleads-US
San Jose, CA
4 days ago
Staff SWE, ML Compiler Performance Engineer Co-Design
$207k - $300k
...Staff SWE, ML Compiler Performance Engineer Co-Design Share Staff SWE... ..., large-scale systems data analysis, visualization... ...), or accelerator kernel programming (Pallas,... ...design, compilers, AI research, and cloud... ...planning, and kernel optimization. Adapt emerging open...
Worldwide
Jobleads-US
Mountain View, CA
1 day ago
Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs
$193.3k - $261.5k
...building Generative AI on AWS. The... ...delivers best-in-class ML inference performance... ...includes an ML compiler, runtime and natively... ...including silicon engineering, hardware design... ...team works to optimize the performance of... ...new and existing systems experience- 5+ years...
Internship
Local area
Work from home
Relocation
Flexible hours
Amazon
Cupertino, CA
4 days ago
Machine Learning - Compiler Engineer , AWS Neuron, Annapurna Labs
$165.2k - $223.6k
...you want to be part of AI revolution? At Amazon our... ...Neuron is the SDK that optimizes the performance of complex ML models executed on AWS Inferentia... ...role is for a software engineer in the Compiler team for AWS Neuron. As... ...) of new and existing systems experience- Experience...
Internship
Local area
Flexible hours
Amazon
Cupertino, CA
1 day ago
ML Runtime Optimization Engineer
$159.05k - $199.3k
...is powering the future of physical AI. Founded in 2017 and now valued at... ...tools and infrastructure, operating systems, and autonomy. Eighteen of the top... ...role We are looking for a software engineer with deep experience in optimizing ML models and deploying them on...
Full time
For contractors
For subcontractor
Casual work
Work at office
Remote work
Day shift
Applied Intuition
Sunnyvale, CA
1 day ago
Machine Learning Engineer -- GPU Kernel
...the next generation of AI builders, and drive transformative... ..., data scientists, and engineers, tackling the most... .... The Role The GPU Kernel Engineer will play a role at the forefront of optimizing performance for the... ...new and cutting-edge systems. The ideal candidate will...
Institute of Foundation Models
Sunnyvale, CA
2 days ago
ML Compiler Engineer for AWS Neuron
...Amazon is seeking a Software Development Engineer for the AWS Neuron compiler team at Annapurna Labs. You will design and implement compiler optimizations to transform ML models for AWS Inferentia and Trainium, collaborating with chip architects and ML teams to deliver...
Jobleads-US
Cupertino, CA
1 day ago
Senior ML Inference & Compiler Engineer
...Corporation is seeking a Senior Machine Learning Applications and Compiler Engineer to advance end-to-end inference optimization across NVIDIA platforms. You will work at the intersection of large-scale systems, compilers, and deep learning to map neural network workloads...
Jobleads-US
Santa Clara, CA
5 days ago
Staff ML Systems Engineer — TPU Performance & Gemini
...Google is seeking software engineers for the Core ML team to advance ML workloads on TPU systems, including JAX and PyTorch, impacting... ...contribute to performance optimization across large-scale ML projects... ...efficient software for massive-scale AI workloads. #J-18808-Ljbffr...
Jobleads-US
Sunnyvale, CA
5 days ago
Senior Machine Learning Applications and Compiler Engineer, LPX
$152k - $241.5k
...Learning Applications and Compiler Engineer!NVIDIA is seeking... ...develop algorithms and optimizations for our LPX inference... ...intersection of large-scale systems, compilers, and deep... ...at top tier ML, compiler, and computer... ...Experience with large-scale AI distributed inference...
Full time
Nvidia
Santa Clara, CA
1 day ago
(USA) Distinguished, Software Engineer-AI/ML Engineer - Agentic Systems & Site Reliability Engineering
$169k - $338k
...WalmartBusiness Segment: Home OfficeAI/ML & Agentic Systems Technical Leadership:... ...develop advanced agentic AI systems that can... ...handle complex reliability engineering workflows, predictive failure analysis, and self-optimization across all Walmart technology...
Full time
Temporary work
Part time
Walmart
Sunnyvale, CA
3 days ago
Senior ML Engineer, Metropolis
$184k - $287.5k
...boundless capabilities of AI to build the next era of... ...automation, and autonomous systems at scale. We seek a Senior ML Engineer to compose and deliver... ...and real-time inference optimization for production deployments... ...Triton, or low-level GPU kernel development for inference...
Full time
Nvidia
Santa Clara, CA
2 days ago
Senior Machine Learning Systems Engineer
....About the Central AI OrgOur organization... ...Conversational AI system that integrates seamlessly... ....About the AI & ML Platform TeamOur... ...a Senior ML System Engineer on the AI & ML Platform... ...will design and optimize large-scale model... ...optimizations (GPU kernels, quantization, speculative...
Work at office
Local area
Atlassian
Mountain View, CA
16 hours ago
Staff ML Systems Engineer
$171.7k - $303.9k
...understand the world!The Data Labeling Engineering team designs, builds, and operates hybrid... ...software engineering, data engineering, and AI/ML, defining the strategies, tooling, and... ..., and direct impact across teams and systems that unblock the next generation of AV capabilities...
Full time
Local area
Remote work
Work from home
Relocation package
Flexible hours
General Motors
Sunnyvale, CA
2 days ago
On-Device ML Infrastructure Engineer (CoreML Runtime), Graphics, Games and Machine Learning
$184.7k - $324.8k
...On-Device ML Infrastructure Engineer (CoreML Runtime), Graphics... ...Machine Learning and AI Imagine being... ...to embedded systems, developing optimization toolkits for model... ...acceleration, building ML compilers and runtimes for... ..., runtime, and kernels for delivering ML...
Relocation
Jobleads-US
Cupertino, CA
1 day ago
Senior Machine Learning Engineer - Predictive World Model
$174.72k - $295.68k
...integrating advanced AI and autonomous driving... ...seeking Machine Learning Engineers with strong expertise... ...-scale deep learning systems, along with solid software... ..., algorithms, code optimization and large-scale data... ...mixed precision, torch.compile, kernel-level optimization, and...
Full time
XPENG Motors
Santa Clara, CA
2 days ago
Sr. AI/ML Platform Engineer
...computing experiences—from AI and data centers, to... ...gaming and embedded systems. Grounded in a... ...:We are hiring AI / ML Platform Engineers to build the... ...Blueprint framework across kernel optimization, RTL/PPA optimization... ...integrations with compilers, ROCm/HIP tooling, profilers...
AMD
Santa Clara, CA
2 days ago
AI/ML Physical Design Flow Engineer
$100k
...industry on cutting-edge AI technology,... ...in software models, compilers, platforms, networking... ...an Physical Design Engineer to lead cross-... ...integrate, and deploy AI/ML-driven solutions... ...turnaround time, and QoR.Optimize EDA tools and... ...access to information, systems, or technologies...
Permanent employment
Tenstorrent
Santa Clara, CA
3 days ago
ML Software Engineer, Data Plane
$165.2k - $223.6k
...Software Development Engineer to own the design... ...performance compute kernels, enabling efficient... ...who can write and optimize low-level code for... ...kernels for a custom ML accelerator... ...architecture, operating systems, and parallel computing... ..., TPU or other AI acceleration hardware...
Local area
Flexible hours
Amazon
Cupertino, CA
16 hours ago
Machine Learning Engineer, AI Inference Solutions (Early in Career)
$119.25k - $150.85k
...mission is to build the ML deployment... ...predictable, and to optimize models so they meet... ...and we’re hiring engineers to help deliver the... ...our sister teams (kernels, compiler, reduced precision... ..., operating systems, computer architecture... ...-on experience in AI/ML (e.g., machine...
Full time
Internship
Local area
Work from home
Relocation package
Flexible hours
General Motors
Sunnyvale, CA
3 days ago
Sr. Machine Learning Engineer, ML Systems Evaluation Engineering
$175k - $308.5k
...United States Machine Learning and AI Join the team redefining... ...that evaluates Siri and AI/ML models at scale to delight and... ...you are excited by large-scale systems, rigorous statistical evaluation... ...Senior Machine Learning Engineer specializing in Conversational...
Relocation
Jobleads-US
Cupertino, CA
5 days ago
Machine Learning Research Engineer
$2,000 per month
...Etched is building AI chips that are... ...image generation Optimize model... ...you have An ML Research background... ...also have ML Systems Research and HW Co... ...Familiarity with GPU kernels, the CUDA compilation stack and related... ...and greatly value engineering skills. We do not...
Work at office
Relocation package
ETCHED LLC
Cupertino, CA
1 day ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Systems ML Engineer - AI Compiler & Kernel Optimizations. Be the first to apply!