"kernel fusion" Jobs

6 open tech roles matching “kernel fusion”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: CUDA, Triton, JAX. Every listing is re-checked daily and closed roles are removed.

Showing 6 of 6 results

Staff Software Engineer - GenAI Performance and Kernel

Databricks San Francisco, California $190.9k–$232.8k/yr Published 11 months ago

Triton oneDNN CUTLASS CUDA NVProf

Apply →

Inference Engineering, Co-op

Join Inferact as a co-op student to work on cutting-edge AI inference systems in a hands-on engineering role.

Inferact San Francisco Published 2 days ago

C++ Triton JAX GPU programming CUDA

Flexible on stack

Apply →

Member of Technical Staff, TPU Performance Engineering Hybrid Visa

Join Inferact as a TPU performance engineer to optimize vLLM for Google TPUs, enhancing AI inference performance.

Inferact San Francisco $200k–$400k/yr Published 2 months ago

JAX vLLM ML Pallas TPU

Flexible on stack

Apply →

GPU Performance Engineer

Join Genmo as a GPU Performance Engineer to optimize video generation models and achieve significant performance improvements.

Genmo San Francisco HQ Published 1 year ago

C++ Nsight Systems Triton CUDA NVProf

Flexible on stack

Apply →

Performance Engineer (Inference, Training & GPU)

Join World Labs as a Performance Engineer to optimize AI models for speed and efficiency in a cutting-edge research environment.

World Labs San Francisco $200k–$300k/yr Published 4 months ago

FP8 C++ Triton JAX CUDA

Flexible on stack 70% coding

Apply →

Performance Engineer, GPU Hybrid Visa

Anthropic San Francisco, CA | New York City, NY | Seattle, WA $280k–$850k/yr Published 11 months ago

FP8 NVLink Triton JAX CUTLASS

Apply →