Machine Learning Engineer – Kernels

Job not on LinkedIn

🕒 May 21

Apply Now
Find Similar Remote Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Logo of Mindbeam AI

Mindbeam AI

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.

📋 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

Similar Jobs

🕒 May 20

AssetWorks Inc

5001 - 10000

☁️ SaaS

🏢 Enterprise

🚗 Transport

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.

Azure

Cloud

Microservices

Postgres

Python

SQL

.NET

🕒 May 20

AssetWorks Inc

5001 - 10000

☁️ SaaS

🏢 Enterprise

🚗 Transport

AI/ML Engineer leading technical implementation of generative AI systems for enterprise, designing multi-agent systems and overseeing AI workflows, ensuring AI implementations are scalable and secure.

Azure

Cloud

Microservices

Postgres

Python

SQL

.NET

🕒 May 17

Paramount

10,000+ employees

📱 Media

👥 B2C

Machine Learning Engineer optimizing onboarding and re-entry experiences within Paramount's streaming services. Focusing on user commitment and personalization through advanced machine learning techniques.

Google Cloud Platform

PyTorch

Tensorflow

🕒 May 15

Camus Energy

11 - 50

⚡ Energy

🤖 Artificial Intelligence

☁️ SaaS

Machine Learning Engineer advancing forecasting and predictive modeling in clean energy software platform. Owning the full lifecycle of ML model development with strong product influence.

Pandas

Python

PyTorch

Scikit-Learn

🕒 May 14

Foundation EGI

11 - 50

🤖 Artificial Intelligence

☁️ SaaS

🏢 Enterprise

ML Ops Engineer at a startup utilizing AI, physics simulation, and computer graphics. Focusing on reducing costs and improving engineering productivity in design and manufacturing.

AWS

Cloud

Docker

Kubernetes

Python

Terraform