"gemm" Jobs

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

Showing 4 of 4 results

Member of Technical Staff, AMD GPU Performance Engineering Hybrid Visa

Join Inferact as an AMD GPU performance engineer to optimize vLLM for the AMD accelerator ecosystem.

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

Triton AMD vLLM CK GPU

Flexible on stack

Apply →

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 →

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 →

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 3 days ago

C++ Triton JAX GPU programming CUDA

Flexible on stack

Apply →