Jobs at SF Tensor

Founding Research Engineer focused on AI-driven compilation at SF Tensor. This role involves designing and implementing reinforcement learning systems and agentic compilation systems that use LLMs to optimize machine learning code. Responsibilities include developing reward models, training infrastructure, and creating representations of compiler IR for learned optimization.

Founding Product Engineer at SF Tensor responsible for the entire user-facing surface of the training platform. This role involves owning product areas from scoping to iteration, building web interfaces with Next.js, backend services in Bun/TypeScript, and managing infrastructure on AWS using Kubernetes.

Founding GPU Compiler Engineer at SF Tensor responsible for building the core compilation infrastructure for their AI compiler. This involves taking models from PyTorch, JAX, and TensorFlow and optimizing them into binaries for large-scale AI pre-training across various hardware targets.

Founding GPU Kernel Engineer at SF Tensor. This role involves deep optimization of GPU kernels for ML workloads, profiling at the microarchitectural level, and translating hand-optimization insights into automated compiler passes. The engineer will focus on maximizing FLOPs and understanding hardware limitations.