Kernel Engineer Core Ml Ml Runtime Jobs in San Francisco, CA

You will work across models, kernels, compilers, and silicon to shape how ML workloads map to hardware—and how the hardware should evolve to support them.

You’ll work on the core ML systems that turn 3D geometry and synthetic data into reliable, production-grade vision models deployed on factory floors.

Experience with ML compilers is a strong positive, but not required.

…Do Design and implement custom ML kernels (e.g., CUDA, CuTe, Triton) for core LLM operations such as attention, matrix

The role At Mach9, ML Engineers build the perception models at the core of our AI-enabled CAD system.

About the Team Training Runtime designs the core distributed machine-learning training runtime that powers everything from early research

Strong understanding of kernels, sharding, runtime systems, or distributed scaling techniques.

Have built or optimized runtimes, distributed systems, compilers, kernels, model-serving infrastructure, or adjacent systems software.

…pass, from model code through kernels to execution on the hardware.

…AITER), apply to AMD GPU Kernel Engineering. For NVIDIA GPUs and cross-platform kernel work, apply to Kernel Engineering.

…are larger than a single- engineer effort, typically spanning multiple engineers and lasting multiple quarters.

…is looking for a Software Engineer to join the ML Developer Experience (MLDevX) team.

…of Technical Staff for GPU Kernel Engineering to establish and push the envelope on what the hardware can actually do before any search

…Points If You Have Deep ML systems background (e.g., training compilers, runtime optimization, custom kernels