1,245 open positions
Senior Software Engineer, Performance, SOC Architect↗
GoogleSunnyvale, CA, US
$174k–$174k5d
Drive continuous improvements to machine learning software/hardware stacks by providing insightful performance debugging for workloads and custom kernels.
Software Engineer II - AI/ML, Neuron Inference
AmazonCupertino, California, United States
6d
Design, develop, and optimize machine learning models and frameworks for deployment on custom ML hardware accelerators (AWS Trainium), focusing on distributed inference and high-performance kernels.
Senior Software Engineer, Infrastructure, Platforms and Devices↗
GoogleSan Francisco, CA, US
$174k–$174k6d
Building large-scale infrastructure, distributed systems, and networks for Google's XR (AI + Extended Reality) devices and platforms.
Staff Software Engineer, ML Performance, GPU↗
GoogleSunnyvale, CA, US
$207k–$207k6d
Optimizing, modeling, and evaluating GPU systems for internal and external ML workloads, including LLM training and serving benchmarks.
Lead Software Engineer, Kernels
MatXMountain View
$160k–$160k7d
Lead a team of kernel engineers to design and implement high-performance compute kernels for specialized AI hardware, partnering with ML and hardware teams to co-design next-generation architectures.
Performance Modeling Engineer
MatXMountain View (HQ) or Remote
$120k–$120k7d
Translate AI accelerator architecture into performance models and maintain modeling frameworks for frontier LLMs.
Machine Learning Performance Engineer, Annapurna Labs
AmazonTel Aviv-Yafo, Tel Aviv, Israel
7d
Building AWS Neuron performance engineering team to profile and optimize deep learning workloads on Inferentia and Trainium chips.
Senior Performance Engineer, Efficiency Red Team
AmazonToronto, Ontario, Canada
$151k–$151k7d
Senior Performance Engineer leading deep technical research to discover and eliminate waste across the entire compute stack (CPU, DRAM, GPU, storage, I/O, networking) at Amazon's scale, turning findings into billions in capacity savings.
Sr. Software Development Engineer, AWS Neuron SDK- Containers
AmazonSeattle, Washington, United States
$168k–$227k7d
Develop and maintain high-performance monitoring and profiling tools (Neuron Profiler) for AWS Inferentia and Trainium ML accelerators to optimize AI workloads and ML frameworks.
← Select a job to preview