"flash attention" Jobs

2 open tech roles matching “flash attention”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: CUDA, NVLink, C++. Every listing is re-checked daily and closed roles are removed.

Showing 2 of 2 results

Software Engineer - GPU Kernels

Join Baseten as a GPU Kernel Engineer to optimize high-performance GPU kernels for cutting-edge AI applications.

baseten San Francisco Published 1 year ago

C++ Nsight Systems Torch Profiler CUDA Machine Learning

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 →