1,062 open positions
Software Engineering Manager, ML Kernel Performance, AWS Neuron, Annapurna Labs
AmazonToronto, Ontario, Canada
$171k–$286k1mo
Architect and implement high-performance compute kernels for ML operations on AWS custom accelerators (Inferentia/Trainium), optimizing deep learning and GenAI workloads at the hardware-software boundary.
Staff Software Engineer, ML Compilers, TPU↗
GoogleNew York, NY, US
$207k–$207k1mo
Building the Accelerated Linear Algebra (XLA) compiler to enable Tensor Processing Units (TPUs) to accelerate machine learning and scientific computing workloads for internal and external customers.
Staff Software Engineer, ML Systems Co-Design↗
GoogleMountain View, CA, US
$207k–$207k1mo
Define strategy and methodology to enable open-source ML models to achieve TPU performance, analyzing mapping gaps and optimizing models ahead of silicon pilot.
ML Software Engineer, Data Plane
AmazonTel Aviv-Yafo, Tel Aviv, Israel
1mo
Design and implement the inference data plane for large language models running on custom hardware, covering model execution, memory management, and data movement.
Lead SDE C/C++ Hardware/Software Co-Design, Machine Learning Acceleration Systems
AmazonCupertino, California, United States
$193k–$262k1mo
Develop bare metal firmware and custom hardware for AWS ML accelerator chips to enable neural network model execution.
← Select a job to preview