Senior ML Compiler Architect: PyTorch to CUDA & MLIR
Jobleads-US
Memwize is a VC-backed stealth-mode startup building rack-level AI inference systems in Sunnyvale, CA. We are seeking a Senior/Principal Machine Learning Compiler Developer to design and implement compiler capabilities from PyTorch through Triton down to CUDA and machine IR on leading-edge accelerators.
Join a fast-moving team focused on performance, scalability, and low-latency inference for next-generation AI workloads.
#J-18808-Ljbffr Jobleads-USVacancy posted 1 day ago
Similar jobs that could be interesting for youBased on the Senior ML Compiler Architect: PyTorch to CUDA & MLIR in Sunnyvale, CA vacancy
$152k - $241.5k
...company”.NVIDIA is hiring a Senior AI Compiler Engineer. GPUs are driving rapid... ...this team, you’ll build an MLIR-based AI compiler that powers... ...algorithms, and frameworks such as PyTorch and JAX.Experience with GPU... ...in GPU architecture with CUDA or OpenCL programming experience...SeniorFull timeRemote workNvidia
Santa Clara, CA4 days ago- ...NVIDIA Corporation is seeking a Senior Machine Learning Applications and Compiler Engineer to advance end-to-end inference optimization... ...(C/C++/Rust), compiler development, and MLIR/LLVM, with experience in TensorFlow, PyTorch, and ONNX. #J-18808-Ljbffr Jobleads-USSenior
$184k - $287.5k
...the worldWe are looking for an experienced Compiler Optimization Engineer for an exciting... ...We deliver features and improvements to CUDA and other compute compilers to better realize... ...for performance improvements in the MLIR/LLVM based compiler middle end optimizer....SeniorFull timeNvidia
Santa Clara, CA1 day ago$152k - $241.5k
...are looking for an experienced Compiler Infrastructure Engineer to join... ...projects such as LLVM, Clang, and MLIR, while building the tooling,... ...automationPractical experience applying AI or ML-based tools to improve... ...in GPU programming, CUDA, or parallel programming modelsFamiliarity...SeniorFull timeRemote workNvidia
Santa Clara, CA4 days ago$152k - $241.5k
...impact on the world.We are hiring software engineers for the CUDA Tile team. NVIDIA GPUs are at the center of the deep... ...major addition to CUDA ( You will design and implement compiler transformations, develop MLIR-based dialects and lowering passes, and optimize the performance...SeniorFull timeRemote workNvidia
Santa Clara, CA1 day ago$152k - $241.5k
...now looking for a Senior Machine Learning Applications and Compiler Engineer!NVIDIA is... ...closely with hardware architects and design teams to... ...at top tier ML, compiler, and computer... ...Experience with LLVM and/or MLIR, including building... ...as TensorFlow and PyTorch, and experience...SeniorFull timeNvidia
Santa Clara, CA1 day ago$184k - $287.5k
...at the intersection of CUDA and Deep Learning Systems... ...silicon prototyping.Architect and optimize distributed... ...architects, kernel and compiler authors and CUDA driver... ...inference frameworks (e.g., PyTorch, JAX, TensorRT, vLLM,... ...learning compilers and ML systems, including...SeniorFull timeNvidia
Santa Clara, CA3 days ago$152k - $241.5k
...are now looking for a Senior Performance Architect for Nemotron! At... ...framework development, compiler, and hardware teams to... ...Solid understanding of ML fundamentals, model... ...learning frameworks like PyTorch, TRT-LLM, VLLM,... ...with GPU computing (CUDA)Your base salary will...SeniorFull timeNvidia
Santa Clara, CA2 days ago- ...build accelerators for high‑demand workloads and contribute to a PyTorch backend prototype for these accelerators. Responsibilities... ...and validation on RTL simulations, with collaboration across AI/ML teams to scale data centers efficiently. #J-18808-Ljbffr Jobleads...Senior
- ...building a cutting-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... ...with x86/FPGA. You will work across ML frameworks, MLIR, and runtime components, collaborating with...Senior
$184k - $287.5k
...center of this platform are CUDA Core Libraries that... .... We are hiring a Senior Software Engineer to develop... ...with C/C++, Python, compiler, and runtime engineers.... ...stacks (CUDA C++/Python, PyTorch, JAX, Numba, CuPy, or similar... ...LLVM, Clang, or MLIR.Experience designing safe...SeniorFull timeNvidia
Santa Clara, CA2 days ago$184k - $287.5k
...center of this platform are CUDA Core Libraries that... ....We are hiring a Senior Software Engineer to advance... ...Collaborate with C/C++, Rust, compiler, and runtime engineers... ...(CUDA C++/Python, PyTorch, JAX, Numba, CuPy, or... ...including LLVM, Clang, or MLIR.Expertise in designing...SeniorFull timeNvidia
Santa Clara, CA1 day ago$184k - $287.5k
...center of this platform are CUDA Core Libraries that... ....We are hiring a Senior Software Engineer to advance... ...Rust.Balance performance, compile time, portability,... ...stacks (CUDA C++/Python, PyTorch, JAX, Numba, CuPy, or similar... ...LLVM, Clang, or MLIR.Knowledge of binary interfaces...SeniorFull timeNvidia
Santa Clara, CA1 day ago$152k - $241.5k
...be doing:In this role, develop compiler optimization algorithms for... ...open-source compilers such as MLIR, LLVM and OpenAI Triton.Designing... ...and distributed programming.CUDA or OpenCL programming experience... ...frameworks such as JAX, PyTorch or TensorFlow.Extensive experience...SeniorFull timeRemote workNvidia
Santa Clara, CA4 days ago- ...motivated and technically exceptional PMTS AI/ML Compiler Engineer to join AMD's AI Software... ...transformations and operator lowerings within MLIR, LLVM, and the ROCm software stack.... ...GPU programming technologies such as HIP, CUDA, Triton, or equivalent parallel programming...Senior
AMD
San Jose, CA4 days ago $184k - $287.5k
...impact on the world.We are seeking an AI Compiler Engineer with deep expertise in compiler technologies... ....8+ years of software engineering and AI/ML experience, preferably in tools or systems... ....Hands-on experience with LLVM/MLIR-based compiler development, including optimization...SeniorFull timeRemote workNvidia
Santa Clara, CA1 day ago$184k - $287.5k
...Learning engineer to bring advanced CUDA features and Distributed... ...into AI stacks, including PyTorch, TRT-LLM, vLLM, SGLang, JAX,... ...drive improvements in the AI Compiler-Runtime interface to build speed... ...AI researchers, HW and SW architects, kernel and compiler authors...SeniorFull timeNvidia
Santa Clara, CA1 day ago$184k - $287.5k
...openings for a Deep Learning Communication Architect. We scale the DNN models and training/... ...training and Inference frameworks, such as PyTorch, TensorRT-LLM, vLLM, SGLang.Strong... ...Familiarity with GPU computing, including CUDA and OpenCL, and familiarity with InfiniBand...SeniorFull timeWork experience placementNvidia
Santa Clara, CA1 day ago- ...DescriptionPrimary Function of Position We are seeking an experienced Senior Insights Architect for Analytics and AI Solutions to lead the design and... ...experience collaborating during building and deploying AI/ML models and GenAI-powered solutions, and experience leading...SeniorLocal areaWorldwideFlexible hours
Intuitive Surgical
Sunnyvale, CA2 days ago $119.8k - $234.7k
...Teams and Xbox Live. We are seeking a Senior AI Performance Architect to help shape the next generation of... ...programming environment such as C++, CUDA, Triton, ROCm/HIP, or an equivalent... ...PCIe or other high-speed I/O, kernels, compilers, and runtimes. Experience...SeniorOngoing contractWork at officeLocal areaWorldwide3 days per weekMicrosoft
Mountain View, CA6 hours ago$152k - $241.5k
...looking for an AI & Deep Learning Compiler Engineer. NVIDIA is hiring... ....Strong programming skills in CUDA including analyzing and debugging... ...compiler technologies (e.g., MLIR, LLVM, XLA, Triton, etc.).... ...algorithms, and frameworks, such as PyTorch, JAX.GPU kernel authoring and...SeniorFull timeRemote workNvidia
Santa Clara, CA1 day ago$183k - $247.6k
...world.We are seeking an experienced Power Architect to join our Machine Learning Acceleration... ..., and managing thermal constraints for ML accelerator designs in Amazon data centers... ...celebrates knowledge-sharing and mentorship. Our senior members enjoy one-on-one mentoring and...SeniorLocal areaFlexible hoursAmazon
Cupertino, CA1 day ago$152k - $241.5k
...looking for an AI & Deep Learning Compiler Engineer. NVIDIA is hiring... ...compiler technologies (e.g., MLIR, LLVM, XLA, Triton, etc.).Excellent... ...CPU and/or GPU architecture. CUDA or OpenCL programming... ...algorithms and frameworks, such as PyTorch, JAX.GPU kernel authoring and...SeniorFull timeRemote workNvidia
Santa Clara, CA1 day ago$198.2k - $297.2k
...with us!Role and ResponsibilitiesAs a Senior Staff GPU Architect - Machine Learning, you will help lead... ...development of innovative machine learning (ML) solutions for Samsung’s premium... ...frameworks and tools (e.g., TensorFlow, PyTorch)Ability to write clear and concise...SeniorHourly payFull timeRelocationSamsung Semiconductor
San Jose, CA4 days ago$163k - $236k
...subsystems.Collaborate with partners in Hardware Design, Software, Compiler, ML Model and Research teams for hardware/software co-design.Work... ...with deep learning frameworks including TensorFlow and PyTorch.Knowledge of Machine Learning market, technological and business...WorldwideGoogle
Sunnyvale, CA2 days ago$193.3k - $261.5k
...performance of complex ML models executed on... ...role is for a senior software engineer in the Compiler team for AWS Neuron.... ...ML frameworks (e.g, PyTorch, TensorFlow, and JAX... ...OpenXLA, StableHLO, MLIR will be added bonus!... ...side by side with chip architects, runtime/OS engineers...SeniorLocal areaFlexible hoursAmazon
Cupertino, CA15 hours ago- ...Google LLC in Sunnyvale, CA, is seeking a Staff ASIC Power Engineer for ML accelerators. You will shape AI/ML hardware acceleration by driving power-aware TPU architectures and implementing robust power models. You will lead power optimization efforts across pre- and...Senior
- ...General Motors LLC seeks a Senior ML Infrastructure engineer to design and scale robust compute platforms for Simulation workflows. You will focus on GPUs, scaling, and reliability while shaping the architecture and roadmap for GM's AI Validation / Simulation needs....Senior
- ...ServiceNow seeks an Agentic AI Harness Architect for Moveworks to define the technical vision and build robust AI-driven agent systems within... ...performance in production environments. Strong leadership and ML expertise are essential, with 8+ years in related roles. #J-18...Senior
- ...NVIDIA seeks a Senior Process Architect to consolidate DriveOS development processes for autonomous mobility software. You will define risk-based... ...standards, you will drive governance, traceability, and AI/ML safety integration, delivering scalable, #J-18808-Ljbffr...Senior
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Senior ML Compiler Architect: PyTorch to CUDA & MLIR. Be the first to apply!
Related searches
- senior mulesoft developer Sunnyvale, CA
- senior business manager Sunnyvale, CA
- senior linux systems engineer Sunnyvale, CA
- senior mainframe developer Sunnyvale, CA
- senior cloud security engineer Sunnyvale, CA
- senior level Sunnyvale, CA
- senior project manager Sunnyvale, CA
- senior network engineer remote Sunnyvale, CA
- senior manager product development Sunnyvale, CA
- senior content marketing manager Sunnyvale, CA