1,047 open positions
Lead Software Engineer, Kernels
MatXMountain View
$160k–$160k4d
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.
Senior Performance Engineer, Efficiency Red Team
AmazonToronto, Ontario, Canada
$151k–$151k4d
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.
Machine Learning Performance Engineer, Annapurna Labs
AmazonTel Aviv-Yafo, Tel Aviv, Israel
4d
Building AWS Neuron performance engineering team to profile and optimize deep learning workloads on Inferentia and Trainium chips.
Staff Software Engineer, ML Systems Co-Design↗
GoogleMountain View, CA, US
$207k–$207k5d
Define and own strategy for enabling open-source ML models to achieve TPU performance; establish scalable methodology to analyze mapping gaps across model families.
Software Development Engineer - Expert Consultant, AGI - Data Services
AmazonBellevue, Washington, United States
$144k–$144k5d
Strategic oversight of human-in-the-loop and model-in-the-loop data pipelines powering next-generation AI solutions, ensuring data excellence from end to end.
Software Engineer
AmazonUSA
8d
Develop hardware abstraction layers (HALs) for new SoC intellectual property and solve system-level architectural issues for AWS machine-learning server infrastructure.
System Software Engineer, Linux Kernel and Device Drivers
MatXMountain View
$160k–$160k9d
Architecting and implementing Linux kernel drivers and low-level userspace libraries for MatX's AI accelerators to enable efficient GenAI silicon.
Software Engineer - Hardware Abstraction Layer, AWS Machine Learning Accelerators
AmazonCupertino, California, United States
$144k–$144k9d
Develop the hardware abstraction layer (HAL) for AWS custom machine learning system-on-chips (SoCs) like Trainium and Inferentia, managing low-level hardware details for cloud-scale ML infrastructure.
Machine Learning - Compiler Engineer , AWS Neuron, Annapurna Labs
AmazonCupertino, California, United States
9d
Building next-generation compilers to optimize ML models (PyTorch, TensorFlow, JAX) for deployment on AWS Inferentia and Trainium chips.
← Select a job to preview