Machine Learning Engineer – Kernels

Mindbeam AI

. Design and implement custom GPU/accelerator kernels to maximize performance.

Posted 5/21/2026full-timeRemote • California • 🇺🇸 United StatesJuniorMid-Level💰 $150,000 - $190,000 per yearWebsite

About the role

Key responsibilities & impact

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.

Requirements

What you’ll need

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.

Benefits

Comp & perks

🌐 Worldwide ❌ Jobs You've Hidden ⭐️ Saved Jobs ✅ Applied Jobs ✉️ Email Alerts 👤 Account Mindbeam AI Website LinkedIn All Job Openings 1 - 10 employees Founded 2024 🤖 Artificial Intelligence 🤝 B2B Artificial Intelligence
B2B Mindbeam AI is a company building Litespark, an ultra-fast LLM pretraining framework that accelerates training and inference for generative AI applications. Its software improves throughput on existing GPU hardware with zero code changes, is compatible with industry-standard ML frameworks like PyTorch, and claims to reduce pretraining time from months to days while lowering energy consumption and costs (up to ~81% energy savings in cited workloads). Mindbeam targets enterprise and B2B customers seeking scalable, efficient AI infrastructure for large language model development. Machine Learning Engineer – Kernels Job not on LinkedIn 🔥 1 minute ago 🏄 California – Remote 💵 $150k - $190k / year ⏰ Full Time 🟢 Junior 🟡 Mid-level 🤖 Machine Learning Engineer Apply Now Find Hiring Managers Customize resume + cover letter Report problem ☆ Save ☑️ Mark as applied ❌ Hide 📋 Description
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. 🎯 Requirements
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. Apply Now 📊 Check your resume score for this job Improve your chances of getting an interview by checking your resume score before you apply. Check Resume Score Similar Jobs Manager, Forward Deployed Machine Learning Engineering 🔥 4 hours ago Federato 11 - 50 🤖 Artificial Intelligence Website LinkedIn All Job Openings Manager of Forward Deployed Machine Learning Engineering leading AI and ML team delivering customer solutions. Ensuring operational excellence while collaborating across leadership in a fast-paced environment. 🇺🇸 United States – Remote 💵 $220k - $250k / year 💰 $15M Series A on 2022-09 ⏰ Full Time 🟡 Mid-level 🟠 Senior 🤖 Machine Learning Engineer 🦅 H1B Visa Sponsor Machine Learning Engineer 🔥 16 hours ago Revelation Pharma 201 - 500 💊 Pharmaceuticals ⚕️ Healthcare Insurance 🤝 B2B Website LinkedIn All Job Openings Machine Learning Engineer designing, training, and monitoring ML models for healthcare applications. Owning model lifecycle management and collaborating with clinical teams for explainable models. 🇺🇸 United States – Remote 💵 $120k - $145k / year ⏰ Full Time 🟡 Mid-level 🟠 Senior 🤖 Machine Learning Engineer AWS MLOps Engineer 🔥 17 hours ago Guidehouse 10,000+ employees Website LinkedIn All Job Openings MLOps Engineer responsible for designing and implementing platforms for the deployment of machine learning solutions. Collaborating with data scientists and engineers to operationalize models in federal environments. 🇺🇸 United States – Remote 💵 113k - 188k / year 💰 Grant on 2023-02 ⏰ Full Time 🟡 Mid-level 🟠 Senior 🤖 Machine Learning Engineer 🦅 H1B Visa Sponsor Cloud Docker Kubernetes Python AI / ML Engineer 🔥 20 hours ago AssetWorks Inc 5001 - 10000 ☁️ SaaS 🏢 Enterprise 🚗 Transport Website LinkedIn All Job Openings Senior Full-Stack Developer leading the technical realization of generative AI roadmap for asset management solutions. Building tools that enhance expert judgment through sophisticated Agentic workflows. 🇺🇸 United States – Remote ⏰ Full Time 🟢 Junior 🤖 Machine Learning Engineer 🦅 H1B Visa Sponsor Azure Cloud Microservices Postgres Python SQL .NET AI/ML Engineer 🔥 20 hours ago AssetWorks Inc 5001 - 10000 ☁️ SaaS 🏢 Enterprise 🚗 Transport Website LinkedIn All Job Openings AI/ML Engineer leading technical implementation of generative AI roadmap at AssetWorks. Building AI systems that power enterprise solutions for asset management challenges. 🇺🇸 United States – Remote ⏰ Full Time 🟢 Junior 🤖 Machine Learning Engineer 🦅 H1B Visa Sponsor Azure Cloud Microservices Postgres Python SQL .NET View More Machine Learning Engineer Jobs 🌐 Worldwide Built by Lior Neu-ner. I'd love to hear your feedback — Get in touch via DM or support@remoterocketship.com Search Search Jobs by country Search jobs by city Search jobs by job title Search entry-level jobs Search junior-level jobs Search senior-level jobs Search jobs by tech stack Search jobs by contract type Search remote internships Search remote part-time jobs Remote jobs Anywhere in the World Companies Hiring Anywhere in the World Companies Hiring Sales People Anywhere in the World Companies Hiring Software Engineers Anywhere in the World Resources Advice Tips for finding remote jobs Interview questions and answers Resume examples Cover letter examples Post a job Affiliates Privacy policy Terms of service Job board SEO course AI Apply Copilot OpenClaw job finder Jobs by Country Remote jobs anywhere in the world (Worldwide remote jobs) Remote jobs United States Remote jobs Australia Remote jobs Brazil Remote jobs Canada Remote jobs France Remote jobs Ireland Remote jobs Germany Remote jobs Netherlands Remote jobs Spain Remote jobs UK Popular Jobs Remote data analyst jobs Remote customer support jobs Remote executive assistant jobs Remote marketing jobs Remote product designer jobs Remote product manager jobs Remote project manager jobs Remote recruiter jobs Remote sales jobs Remote software engineer jobs Jobs by Type Remote full-time jobs Remote part-time jobs Remote contract jobs Remote internship jobs Remote entry-level jobs Remote jobs with no experience required Remote junior jobs (1-3 years of experience) Digital nomad jobs Remote jobs with no degree required Freelance remote jobs Temporary remote jobs Remote jobs hiring now Stay at home mom jobs

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

GPU programmingparallel computingsystems-level optimizationC++CUDAML frameworksperformance diagnosticsprofiling toolslow-level optimizationdistributed computing

Certifications

Bachelor’s degreeMaster’s degreePhD