Software Engineer, Systems ML - Compilers / Kernels
Jobleads-US
- Software Engineer, Systems ML - Compilers / KernelsSanta Clara, CA Engineering+ 2 moreApply nowIn this role, you will be a member of the MTIA (Meta Training & Inference Accelerator) Software team and part of the bigger PyTorch AI framework organization. MTIA Software Team has been developing a comprehensive AI Compiler strategy that delivers a highly flexible platform to train & serve new DL/ML model architectures, combined with auto-tuned high performance for production environments across specialized hardware architectures. The compiler stack, DL graph optimizations, and kernel authoring for specific hardware, directly impacts performance and deployment velocity of both AI training and inference platforms at Meta.You will be working on one of the core areas such as PyTorch framework components, AI compiler and runtime, high-performance kernels and tooling to accelerate machine learning workloads on the current & next generation of MTIA AI hardware platforms. You will work closely with AI researchers to analyze deep learning models and lower them efficiently on MTIA hardware. You will also partner with hardware design teams to develop compiler optimizations for high performance. You will apply software development best practices to design features, optimization, and performance tuning techniques. You will gain valuable experience in developing machine learning compiler frameworks and will help in driving next generation hardware software codesign for AI domain specific problems.---## Software Engineer, Systems ML - Compilers / Kernels Responsibilities* Development of the SW stack with one of the following core focus areas: AI compiler stack, frameworks, high-performance kernel development and acceleration onto the next generation of hardware architectures* Contribute to the development of the PyTorch AI framework core compilers to support new state of the art inference and training AI hardware accelerators and optimize their performance* Analyze deep learning networks, develop & implement compiler optimization algorithms
- J-18808-Ljbffr Jobleads-US
Vacancy posted 1 day ago
Similar jobs that could be interesting for youBased on the Software Engineer, Systems ML - Compilers / Kernels in Santa Clara, CA vacancy
$184k - $287.5k
...highly motivated software professional to work... ...and Deep Learning Systems. As the complexity... ..., custom kernel development, and cluster... ...architects, kernel and compiler authors and CUDA... ...Science, Computer Engineering, Electrical... ...learning compilers and ML systems, including...SuggestedFull timeNvidia
Santa Clara, CA3 days ago- ...gaming and embedded systems. Grounded in a... ...critical in enhancing GPU kernels, deep learning... ...utilizing cutting-edge compiler technologies and advanced engineering principles to drive... ...graph compilers. Software Engineering Best Practices... ...compute into ML frameworks (e.g.,...Suggested
AMD
Santa Clara, CA1 day ago - ...advance your career.THE ROLE: We are seeking a Senior Triton Compiler and Kernel Engineer to advance Triton performance and capabilities on AMD GPUs... ...GPU kernels in Python. It is a critical layer in the AI software stack, connecting frameworks and workloads to GPU hardware...Suggested
AMD
San Jose, CA3 days ago - ...gaming and embedded systems. Grounded in a... ...adopted language and compiler for high-performance GPU kernels, powering major AI... ...critical to AMD’s AI software roadmap.AMD GPUs are... .../Gluon kernels for ML kernels powering the... ..., and performance engineering. You are comfortable...Suggested
AMD
San Jose, CA2 days ago - ...and technically exceptional PMTS AI/ML Compiler Engineer to join AMD's AI Software organization. In this role, you... ...collaborating across architecture, kernel, runtime, and framework teams.THE... ...optimizing performance-critical software systems.Proficiency with Linux development...Suggested
AMD
San Jose, CA4 days ago - ...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...
$152k - $241.5k
...technological advancement.Are you a motivated system software engineer with a deep understanding of device... ...Graphs to improve the scheduling of AI/ML workloads on our GPUS to be more... ...and consistency modelsBackground with kernel mode developmentExperience with Linux Systems...Full timeNvidia
Santa Clara, CA2 days ago- ...looking for a motivated engineer to join a core team of... ...machine learning model compiler targeting AMD AIE in Ryzen... ...be passionate about software development. The candidate... ...compiled model with ML Frameworks (such as ONNX... ..., source code control systems (GitHub) and profilers...
AMD
San Jose, CA4 days ago - ...Cerebras Systems is seeking a Kernel Engineer in Sunnyvale to develop high-performance software for ML and HPC workloads on the Cerebras Wafer-Scale Engine. You will implement... ..., and collaborate across kernel, compiler, perf, and hardware teams. This role targets...
$190k - $300k
...the forefront of software and hardware innovation... ...: Staff Software Engineer, SIMD KernelsWhat... ...of the SIMD Kernels team, you will help... ...software kernels for ML operators—such as... ...data structures, system software, and machine... ...working with ML compilers and algorithms, such...Work experience placementRemote workd-Matrix
Santa Clara, CA1 day ago- ...are at the forefront of software and hardware... ...Role: Principal Software Engineer, KernelsWhat you will... ...maintenance of software kernels for next-generation AI... ...will work with a team of compiler experts to build out the... ...with other software (ML and systems) and hardware (mixed signal...Work experience placement3 days per week
d-Matrix
Santa Clara, CA1 day ago $174k - $252k
...and test product or system development code.... ...new and exciting ML paradigms (such as... ...of experience with software development in one... ...PyTorch, or ML compilers such XLA.Proven track... ...Google's software engineers develop the next-... ...workloads and custom kernels. You will provide...Google
Sunnyvale, CA3 days ago- Cerebras Systems builds the world's largest AI chip... ...The RoleThe Core ML team develops novel... ...Cerebras Wafer-Scale Engine. Our work spans efficient... ...ML frameworks, compilers, runtimes, and low-level kernels to implement new algorithmic... ....Contribute to software architecture and...
Cerebras Systems
Sunnyvale, CA4 days ago - ...worlds of semiconductors, AI systems, and software innovation, We’re on a... ...sophisticated mid- and backend compiler techniques to target Persimmons... ...with custom operations and kernels, as well as implement compiler... ...flows from high-level ML representations to low-level...Work experience placementFlexible hours
Persimmons
San Jose, CA18 hours ago $134.5k - $201.7k
...Qualcomm Atheros, Inc.Job Area:Engineering Group, Engineering Group > Software EngineeringGeneral... ...Engineers collaborate with systems, hardware, architecture,... ...WiFi wireless systems, AI/ML-based traffic... ...languages.Understanding of Linux kernel internals and hands-on experience...Work experience placementWork from home$174k - $252k
...Senior Software Engineer, Performance, SOC Architect Share... ...performance, large-scale systems data analysis,... .... ~ Experience with ML frameworks such as TensorFlow... ..., and PyTorch, or ML compilers such XLA. ~ Proven track... ...workloads and custom kernels. You will provide...$165.2k - $223.6k
...Neuron is the complete software stack for the AWS Inferentia... ...Software Development Engineer for the Neuron Runtime... ...bottlenecks and system behavior. Improving performance of ML Kernels and ML Frameworks.In this... ...ensure that the our C++ compiler generates key information...InternshipLocal areaWork from homeFlexible hoursAmazon
Cupertino, CA2 days ago$193.3k - $261.5k
...builds AWS Neuron, the software development kit used... ...accelerator. Neuron includes an ML compiler, runtime, collectives... ...and high-performance kernels that these methods... ..., and distributed systems, where you will help shape... ..., and collectives engineers to land it. You will translate...InternshipLocal areaWork from homeFlexible hoursShift workAmazon
Cupertino, CA1 day ago$152k - $241.5k
...Machine Learning Applications and Compiler Engineer!NVIDIA is seeking engineers... ...intersection of large-scale systems, compilers, and deep learning... ...and design teams to feedback software observations, influence... ...spatial accelerators at top tier ML, compiler, and computer architecture...Full timeNvidia
Santa Clara, CA1 day ago$207k - $300k
...planning, and XLA/kernel optimization.Adapt... ...under real-world compiler constraints.Partner... ...Compiler, and Cloud AI Systems to steer multi-... ..., and launching software products.5 years of... ...proficiency ML workload performance... ...Google's software engineers develop the next-generation...WorldwideGoogle
Mountain View, CA2 days ago$224k - $356.5k
NVIDIA is hiring engineers to scale up the introduction of next generation architecture... ...introductions (NPIs), distributed systems, familiarity with software testing and deployment, and excellent... ...stand out from the crowd:Developing ML/AI infrastructure. Developing bare metal...Full timeNvidia
Santa Clara, CA18 hours ago- ...design, IP, firmware, software and field teams that own... ...PERSONYou have a passion for system architecture, silicon... ...collaborating with engineers located in different sites... ...BSPs, device trees, kernel and rootfs tuning, and... ...Workflows: Use AI/LLM/ML tools to surface design...
AMD
San Jose, CA4 days ago $165.2k - $223.6k
...builds AWS Neuron, the software development kit... ...and Trainium ML accelerators. This... ...toolkit includes an ML compiler, runtime, and... ...software boundary, our engineers build systematic... ...high-performance kernels for ML functions,... ...across the stack from system level...Work experience placementInternshipLocal areaFlexible hoursAmazon
Cupertino, CA2 days ago- ...forward.THE ROLEAMD is seeking a Senior Staff Software Engineer to lead compiler development and GPU performance optimization... ...to low-level GPU code generation and kernel tuning. You are comfortable solving complex system problems, influencing technical direction across...
AMD
San Jose, CA18 hours ago $129.3k - $194.7k
...Software Engineer, Siri Runtime Systems and Interaction Cupertino, California, United States Software and Services Apple is where individual imaginations... ...of existing systems, debugging, integrating ML models in software pipelines, and collaborating closely...Relocation$169k - $338k
...Segment: Home OfficeAs a Distinguished AI/ML Engineer within Walmart Global Tech’s Reliability... ...of next-generation agentic AI systems and intelligent automation solutions that... ...Organization is built with hybrid systems and software engineers who take technical ownership for...Full timeTemporary workPart timeWalmart
Sunnyvale, CA4 days ago$184k - $287.5k
NVIDIA is seeking a Senior Software Engineer to join our CSP Engagements team, focusing on system software for Datacenter products such... ...in embedded firmware, Linux kernel development, and middleware development... ...applications with focus on AI/ML and HPC workloads.Perform...Full timeNvidia
Santa Clara, CA3 days ago- ...implement, and maintain production GPU system software across kernel drivers, runtimes, firmware... ...Collaborate with silicon, firmware, compiler, library, machine-learning framework... ...platform, validation, and production engineering teams. Improve resilience through...Full time
$152k - $241.5k
...looking for an outstanding Senior Compiler Engineer to help build the next... ...intersection of compilers, agentic systems, numerical correctness, and... ...code transformations across software and hardware boundaries.In... ..., runtimes, libraries, kernels, and GPUs.Collaborate closely...Full timeRemote workNvidia
Santa Clara, CA1 day ago$207k - $300k
Lead the end-to-end software architecture, design, and development... ...with hardware, software, and system engineering teams to drive pre-silicon Firmware... ...solutions—spanning FW, BIOS, Kernel, and drivers—ensuing rigorous... ...compute servers and ML headnodes. You will also contribute...WorldwideGoogle
Sunnyvale, CA3 days ago
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Software Engineer, Systems ML - Compilers / Kernels. Be the first to apply!
Related searches
- software engineer full time Santa Clara, CA
- graduate software developer no experience Santa Clara, CA
- software engineer healthcare Santa Clara, CA
- network software engineer Santa Clara, CA
- software engineer internship Santa Clara, CA
- senior software engineer Santa Clara, CA
- software system engineer Santa Clara, CA
- software developer Santa Clara, CA
- ngo software engineer Santa Clara, CA
- startup software engineer Santa Clara, CA