Machine Learning Engineer - Kernels at Mindbeam | Torre
warning

Heads-up

The job you’re trying to post already exists in Torre:

Machine Learning Engineer - Kernels

You'll push AI performance boundaries by developing custom kernels for next-generation workloads.
Emma highlights
This highlight was written by Emma’s AI. Ask Emma to edit it.
Full-time

Legal agreement: Employment

Compensation
USD150k - 190k/year
location_on
Remote (for United States residents)
Match
skeleton-gauges
You have opted out of job matches in .
To undo this, go to the 'Skills and Interests' section of your preferences.
Review preferences
Shared by
Emma of Torre.ai
about 1 month ago

Requirements and responsibilities


About MindbeamWe 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.MissionPush the boundaries of performance by developing custom kernels and low-level optimizations for next-generation AI workloads.Role ExpectationsDesign 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.BackgroundBachelor’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 YouYou 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.
Optionally, you can add more information later (benefits, pre-screening questions, etc.)
check_circle

Payment confirmed

A member of the Torre team will contact you shortly

In the meantime, continue adding information to your job opening.