"gemm" Jobs

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

Showing 8 of 8 results

Member of Technical Staff, AMD GPU Performance Engineering Visa

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

Inferact Singapore S$200k–S$400k/yr Published 2 months ago

Triton AMD vLLM CK GPU

Flexible on stack

Apply →

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 →

Member of Technical Staff, TPU Performance Engineering Visa

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

Inferact Singapore S$200k–S$400k/yr Published 2 months ago

JAX vLLM ML Pallas TPU

Flexible on stack

Apply →

Senior Software Engineer - AI Compute Libraries & Performance Hybrid

Join Graphcore as a Senior Software Engineer to build high-performance AI compute libraries for next-generation hardware.

Graphcore Bristol, UK Published 1 month ago

fused operations C++ GEMM Agile tensor operations

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, California, United States Published 1 day ago

C++ Triton JAX GPU programming CUDA

Flexible on stack

Apply →

Senior Software Engineer - GPU Kernel Authoring & Optimization

Join CoreWeave as a Senior Engineer to optimize GPU kernels for high-performance AI applications in a rapidly growing environment.

Coreweave Sunnyvale, CA / Bellevue, WA $182k–$242k/yr Published 1 month ago

C++ JAX NVIDIA CUDA MLPerf

70% coding

Apply →