S

SF Tensor

Hardware-agnostic AI and HPC software stack

Member of Technical Staff - GPU Kernels

Systems Software EngineeringConfirmed live in the last 24 hours
Full-timeSan Francisco, CA, USA

About the Job

Requirements

  • A track record of hand-writing kernels that match or beat vendor libraries.
  • Comfort reading PTX, SASS, GCN/CDNA ISA, or equivalent machine-level assembly.
  • Fluency with low-level profiling tools such as Nsight Compute, Nsight Systems, rocprof, or omniperf.
  • Solid systems programming skills in C++ and CUDA or ROCm/HIP, with a working understanding of how high-level machine-learning operations map onto hardware, including where framework layers get in the way.

Responsibilities

  • Write and hand-optimize kernels for real workloads to establish the performance ceiling before the search process explores it.
  • Profile at the microarchitectural level, including SM and CU utilization, warp stalls, memory bank conflicts, register pressure, instruction throughput, and occupancy tradeoffs.
  • Debug clock behavior, thermal throttling, and driver paths, including areas where documentation is missing or inaccurate.
  • Convert optimization knowledge into machine-searchable structures that the compiler can explore autonomously.
  • Work below PTX at the ISA level, reasoning about SASS and cubins to emit schedules that PTX cannot express.
  • Build performance models, microbenchmarks, and tooling to predict kernel behavior.
  • Work with the formal correctness team so aggressive kernels ship with a formal proof.
  • Spend approximately two-thirds of the role on kernels and one-third on translating findings into the compiler search space.

Desired Qualifications

  • Experience with compiler backends, including MLIR, LLVM, code generation, instruction selection, or scheduling.
  • Research in superoptimization, program synthesis, formal verification, or search-based compilation.
  • Experience with silicon beyond NVIDIA, including AMD MI-series, TPU, Trainium, or mobile edge GPUs such as Metal, Mali, or Adreno.
  • Experience with distributed AI training.
  • Experience with high-speed interconnects such as NVLink, NVSwitch, InfiniBand, or RoCE.
  • An HPC background in large-scale scientific computing, MPI, or supercomputing.
  • Experience in driver development or a background in EE, computer architecture, or hardware design.

About the Company

SF Tensor

Hardware-agnostic AI and HPC software stack

SF Tensor builds software and infrastructure for AI and HPC to reduce the infrastructure burden. It offers Emma Lang, a hardware-agnostic language that runs efficiently across GPUs and TPUs without rewriting code, plus the SF Tensor Stack with Kernel Optimizer and Elastic Cloud. Kernel Optimizer transforms models by simulating hardware topology to reach efficient forms, often outperforming hand-tuned code; Elastic Cloud finds the best-priced hardware across clouds and coordinates training from one to thousands of GPUs. Its goal is to make high-performance computing fast, affordable, and portable so AI teams can focus on research instead of infrastructure.

Company size
1-10
Company stage
Seed
Total funding
$130K
Headquarters
San Francisco, CA
Founded
2025

Recently Posted Jobs

Explore more opportunities at SF Tensor

See more jobs