Machine Learning Engineer - Kernels
Full-time
Mindbeam
About Mindbeam
We are building the next-generation AI infrastructure for open source and enterprise. Our work is deeply research-oriented and passionate about developing ground-breaking innovations to take state-of-the-art AI applications to the next level.
Mission
Push the boundaries of performance by developing custom kernels and low-level optimizations for next-generation AI workloads.
Role Expectations
• Design and implement custom GPU/accelerator kernels to maximize performance.
• Profile, benchmark, and optimize critical ML workloads.
• Collaborate with researchers to translate algorithmic advances into efficient, production-ready code.
• Stay current with hardware advancements (CUDA, ROCm, TPU) to inform kernel design.
• Document and share best practices for low-level optimization.
Background
• Bachelor’s, Master’s, or PhD in Computer Science, Electrical Engineering, or related field—or equivalent experience.
• 2+ years of experience in GPU programming, parallel computing, or systems-level optimization.
• Strong coding skills in C++, CUDA, or similar languages.
• Familiarity with ML frameworks and their low-level backends.
• Experience optimizing workloads for distributed and heterogeneous compute environments.
• Comfort with profiling tools and performance diagnostics.
About You
You are detail-oriented, performance-obsessed, and excited by the challenge of squeezing out every ounce of compute efficiency. You enjoy working at the intersection of algorithms and hardware, and you thrive in a collaborative environment where bold ideas are encouraged.
Vacancy posted 22 hours ago
Similar jobs that could be interesting for youBased on the Machine Learning Engineer - Kernels in United States vacancy
$213k - $263k
...our infra follows. We are looking for engineers with ML software & systems expertise to... ...system perspective, from efficient deep learning models, model compression, ML software (... ...them onboard through custom NVIDIA GPU kernel development. Deep dive into the NVIDIA...SuggestedFull timeRemote workWaymo
Remote22 hours ago$165.2k - $223.6k
...development kit used to accelerate deep learning and GenAI workloads on Amazon’s custom machine learning accelerators, Inferentia and Trainium.The Acceleration Kernel Library team is at the forefront... ...hardware-software boundary, our engineers craft high-performance kernels...SuggestedInternshipLocal areaWork from homeFlexible hours$130k - $150k
Find your next opportunity today. Sr. Machine Learning Engineer Phoenix, AZ85083 Employment Type: Perm Job Number: 25827Remote Options... ...include eliminating hardware bottlenecks through CUDA kernel tuning and GPU parallel computing, ensuring deep learning models...SuggestedPermanent employmentRemote work$200k - $260k
...reliability. We're looking for a Senior ML Engineer to drive the model serving layer for... ...across the full inference stack — from GPU kernels to framework-level bottlenecks — and ship... ...a strong plus but not required — you can learn this quickly if you have strong ML...SuggestedFull timeTogether Ai
San Francisco, CA22 hours ago$200k - $220k
...Role: We're looking for a Research / ML Engineer to join our Speech Team to build state-of... ...user-perceived quality. Are eager to learn every day, and to find and solve unique... ...step-count reduction, quantization, and kernel/memory optimization to meet interactive latency...SuggestedFull timeWork at officeFlexible hoursCantina Llc
United States22 hours ago$190k - $270k
...prosthetic (O&P) devices. The Role We're hiring a machine learning engineer to work on our Computational Design / CAD to own the parametric... ...geometry programmatically through CAD/geometry APIs and kernels, moving beyond GUI-based workflows toward high-throughput,...Full timeHike Medical
San Francisco, CA22 hours ago- ...success story, then let’s talk! Role Overview As a Senior Machine Learning Engineer, you will be the person we trust with the training side of... ..., trl). Bonus for JAX; extra bonus for having read a CUDA kernel and not flinched. ~ You’ve already built or seriously...Full time
Evocs
United States22 hours ago $213k - $263k
...support and automate the lifecycle of the machine learning workflow, including feature and... ...and Simulation. We are looking for engineers with ML software & systems expertise to... ...continuous batching). Experience with custom kernel development (e.g., CUDA/CUDA Tile,...Full timeRemote workWaymo
Remote22 hours ago$180k
...highly motivated, and focused on engineering excellence. This organization... ...jobs that continuously learn from product data. Iterate and... ...scalability and efficiency of machine learning systems BASIC QUALIFICATIONS... ...experienced in writing CUDA kernels COMPENSATION AND BENEFITS: $1...Temporary workPantera Capital
Palo Alto, CA1 day ago- ...with MCP/tool calling and frameworks such as LangGraph, Semantic Kernel, or OpenAI Agents SDK. Strong Python development skills for... ...AI/ML architecture and solutions. Work closely with engineering and business teams to build and deploy AI models. Design and...
Neshent Technologies
Westmont, IL22 hours ago - ...Contract Position Overview We are seeking a highly skilled AI Engineer – Agentic AI Development to join an exciting enterprise AI initiative... ...or more of the following: LangGraph AutoGen CrewAI Semantic Kernel Claude Agent SDK OpenAI Assistants API Strong software...Full timeContract work
Nova Web Technologies LLC
Seattle, WA4 days ago - ...agentic framework (e.g., LangGraph/LangChain, LlamaIndex, Semantic Kernel, AutoGen) and tool/function calling patterns. Solid knowledge... ...concepts and at least one vector DB in production. Strong engineering practices: code reviews, testing, telemetry, secure-by-design,...
BeaconFire Inc.
New York, NY3 days ago $215k - $275k
...transformation journeys. We blend design, product engineering, analytics, automation, and AI-native... ...-grade AI systems that combine machine learning, generative AI, agentic workflows,... ...LangChain, LangGraph, LlamaIndex, Semantic Kernel, Haystack, CrewAI, Hugging Face, or comparable...Temporary workLocal areaSlalom
Charlotte, NC9 hours ago$184.7k - $324.8k
...Sr. Machine Learning Engineer, Foundation Models Inference - Cloud OS & Inference Santa Clara, California, United States Machine Learning and AI... ...or Nvidia Triton Server. Experience authoring custom CUDA kernels using CUDA C++ or OpenAI Triton. MS in Computer Science, Machine...WorldwideRelocationApple
Seattle, WA4 days ago- ...cutting-edge technology. Interested in learning more? Find out what it’s like to... .... We combine state‑of‑the‑art machine learning, modern data engineering, and agent‑driven automation to enhance... ...frameworks (Strands, LangChain, Semantic Kernel, CrewAI, AutoGen). Knowledge of...
Financial Independence Group
Cornelius, NC4 days ago $128.7k - $261.3k
...and more accessible mobility.For the AI Kernels & Compilers team, that mission shows... ..., kernel development, and performance engineering so that every cycle on our accelerators... ...developing and deploying machine learning modelsCompensation: The compensation information...Flexible hoursGeneral Motors
Sunnyvale, CA22 hours ago$82k - $123k
...Role description Machine Learning Engineer ML Engineer II Who We Are: Born digital, UST transforms lives through... ...Accelerator / Compute Stack * Understanding of GPU memory, kernel execution, synchronization, device selection and host/device...Full timeTemporary workPart timeWork at officeLocal areaFlexible hoursUST Inc
Portland, OR3 days ago$250k - $350k
...alongside some of the world's leading ML systems engineers, including leaders behind Megatron-LM, SGLang, Liger Kernel, TorchRec, CleanRL, TorchRL, and JAX-MD.We're... ...large-scale training, inference, and reinforcement learning. You'll own critical pieces of the ML systems...Visa sponsorshipPeriodic Labs
Menlo Park, CA4 days ago- ...for Fulltime only. Who You Are You are a Senior AI/ML Full Stack Engineer who brings deep hands-on experience building production-grade... ...agents using frameworks such as LangChain, LlamaIndex, Semantic Kernel, or CrewAI Develop and deploy solutions using AWS Bedrock,...Full timeWork at office
Synapse Tech Services Inc
Plano, TX8 hours ago - ...cloud.We're building a new core group of engineers in TLV (Tel Aviv) to drive innovation... ...systems performance and software. As a Machine Learning Performance Engineer, you'll help shape... ...Annapurna hardwareDeveloping high-performance kernels for critical ML operationsEnhancing the...InternshipFlexible hours
$193.3k - $261.5k
The Product: AWS Machine Learning accelerators are at the forefront of AWS innovation. Trainium... ...which includes an ML compiler, the Neuron Kernel Interface (NKI) compiler, and a runtime... ...scale with the world's most talented engineers. Our team covers multiple disciplines...InternshipLocal areaWork from homeRelocationFlexible hours$174.72k - $295.68k
...transportation through cutting-edge R&D in AI, machine learning, and smart connectivity.We are looking for a full-time Machine Learning Engineer - AI Foundation, with deep knowledge and... ...and high-throughput inference.Optimize kernels and model graphs using tools like CUDA,...Full time$220k - $330k
...closely with other teams from product, engineering, sales, logistics, operations, and... ...success.About the RoleWe are looking for a Machine Learning Engineer to apply the latest research... ...on-device inference and vision kernels across CPU/GPU/NPUTrack record of developing...Full timeWork experience placementLive inImmediate startRemote work- ...or advanced degree in Computer Science, Engineering, or related field. Hands-on experience... ...practices. Nice to Have: CUDA kernel development and model optimization (quantization... ...innovation, execution, and continuous learning. Competitive salary, meaningful...Full time
Tycho Ai
Remote22 hours ago - ...way. The Community You Will Join: Machine Learning and Artificial Intelligence are at the... ...do: As a principal machine learning engineer, you will be responsible for fine-tuning... ...inference, GPU inference, pytorch, kernel development Preferred Qualifications...Remote jobFull timeCasual workLive inWork at office
Airbnb, Inc.
United States22 hours ago $220k - $280k
...class latency and reliability. We're looking for a Staff ML Engineer to drive the model serving layer for voice workloads. You'll work... ...conduct systematic profiling and root-cause analysis from GPU kernel behavior to framework-level bottlenecks; drive shipped...Full timeTogether Ai
San Francisco, CA22 hours ago- Obsidian is seeking an MLOps Engineer to join their AI lab's cutting-edge GenAI team in New York. This full-time role involves training and evaluating advanced AI models using PyTorch and kernel-level programming. The ideal candidate should have 2+ years of experience in...Full time
- ...Help us bridge machine learning research and real-world deployment! MatrixSpace develops AI... ...looking for a hands-on Machine Learning Engineer who enjoys turning cutting-edge ML... ...LangGraph, LlamaIndex, AutoGen, Semantic Kernel, or similar frameworks. Experience optimizing...Full time
MatrixSpace
Burlington, MAa month ago - ...Job Description Job Description Machine Learning Engineer – Computer Vision & Robotics Tycho.AI is redefining the future of autonomous... ...software development practices. Nice to Have: CUDA kernel development and model optimization (quantization, pruning,...
ICONSTAFF
Cambridge, MA9 days ago - ...researchers, data scientists, and engineers, tackling the most... ...performance computing in deep learning, driving impactful discoveries... ...optimizing performance for the machine learning software stacks, especially... ...team to develop appropriate kernels and systems for new model...Work experience placementVisa sponsorship
Institute of Foundation Models
Sunnyvale, CA5 days ago
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Machine Learning Engineer - Kernels. Be the first to apply!
Related searches
- ai ml engineer United States
- graduate machine learning engineer United States
- staff machine learning engineer United States
- junior machine learning research engineer United States
- junior machine learning engineer United States
- senior ml engineer United States
- machine learning ai engineer United States
- lead machine learning engineer United States
- entry level machine learning engineer United States
- computer vision machine learning engineer United States