Member of Technical Staff, Performance Optimization

Job summary

Responsibilities

We're looking for a Software Engineer focused on Performance Optimization to help push the boundaries of speed and efficiency across our AI infrastructure. In this role, you'll take ownership of optimizing performance at every layer of the stack—from low-level GPU kernels to large-scale…

• Optimize system and GPU performance for high-throughput AI workloads across training and inference
• Analyze and improve latency, throughput, memory usage, and compute efficiency
• Profile system performance to detect and resolve GPU- and kernel-level bottlenecks
• Implement low-level optimizations using CUDA, Triton, and other performance tooling
• Drive improvements in execution speed and resource utilization for large-scale model workloads (LLMs, VLMs, and video models)

What they're looking for

• Bachelor’s degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent practical experience
• 5+ years of experience working on performance optimization or high-performance computing systems
• Proficiency in CUDA or ROCm and experience with GPU profiling tools (e.g., Nsight, nvprof, CUPTI)
• Familiarity with PyTorch and performance-critical model execution
• Experience with distributed system debugging and optimization in multi-GPU environments

Tech stack

• PyTorch
• CUDA
• Triton
• Kubernetes