Tech Jobs

Waymo • Mountain View - On-Site

Waymo is hiring a Staff Software Engineer for their ML Frameworks & Efficiency team to optimize distributed ML systems for autonomous driving. You'll work with technologies like JAX and XLA on TPUs and GPUs. This position requires expertise in ML frameworks and systems.

Amazon • Cupertino - On-Site

Amazon is hiring a Senior Machine Learning Engineer for the AWS Neuron Frameworks team to develop and enhance support for PyTorch and JAX. You'll work with AWS's machine learning accelerators and engage with open source communities. This position requires experience in machine learning infrastructure and frameworks.