Back to all jobs
Mindbeam logo

Machine Learning Engineer - Kernels

Mindbeam
United States$150k–190k/yrRemote
Employment
Full-time

About the role

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.

731,000+ hidden jobs like this

Mindbeam and thousands of companies post here first — often days before LinkedIn or Indeed. Your first 5 applications are free; go Pro to apply without limits.

Everything Pro unlocks:

  • Unlimited applications — free stops at 5
  • Track every application in one place
  • Apply straight to the source, one click
  • Save & organize roles you love
  • Roles pulled from company boards before the big sites

Weekly

$9.99
$4.99/week

For an active search. Cancel anytime.

Most popular

Monthly

$24.99
$12.99/month

The smart pick. Save 35% vs weekly.

Lifetime

$99
$49.99once

Pay once. Every future feature, forever.