Kernel Performance Engineer (CUDA/Triton)

General Diffusion, Inc.; San Francisco, Northern (CA, KY); Hybrid; USD 180,000 - 280,000; Full time

Job summary

General Diffusion, Inc. in San Francisco seeks a senior engineer to own kernel hot paths across heterogeneous accelerators, turning profiler evidence into portable, numerically correct kernels.

You will profile workloads, design and tune kernels in CUDA/Triton, optimize data layouts and launch configurations, and maintain regression tests and reference implementations to prove performance and correctness. The role emphasizes portability, clear documentation for compiler and runtime partners, and

Qualifications

• Demonstrated low-level accelerator-kernel engineering in CUDA, Triton, or similar.
• Fluency with profiler-led performance analysis and the ability to connect measurements to the GPU memory hierarchy and execution model.
• A rigorous approach to correctness: reference comparisons, numerical-tolerance policies, and regression gates for optimized code.
• Judgment about when specialization is warranted and how to state a kernel's envelope while retaining reliable fallbacks.
• Ability to communicate compact, reproducible evidence to collaborators, separating kernel findings from broader conclusions.

Responsibilities

• Profile representative workloads to identify kernel bottlenecks and execution regimes.
• Design, implement, and tune priority kernels in a low-level accelerator stack (CUDA, Triton, or equivalent).
• Use hardware counters and experiments to reason about memory traffic, cache behavior, and occupancy.
• Maintain numerical-correctness coverage across shapes, dtypes, layouts, and boundary conditions.
• Maintain versioned microbenchmarks and regression reports documenting environment and baselines.
• Document kernel paths, constraints, and safe fallbacks for compiler/runtime partners.

Skills

Low-level kernel engineering

CUDA

Triton

Profiling & analysis

Numerical correctness

Regression testing

Communication of results

Tools

CUDA

Triton

Job description

General Diffusion, Inc. in San Francisco seeks a senior engineer to own kernel hot paths across heterogeneous accelerators, turning profiler evidence into portable, numerically correct kernels.

You will profile workloads, design and tune kernels in CUDA/Triton, optimize data layouts and launch configurations, and maintain regression tests and reference implementations to prove performance and correctness. The role emphasizes portability, clear documentation for compiler and runtime partners, and