Senior AI/ML Software Engineer - High-Perf Inference

Amazon Web Services (AWS); Seattle (WA); Presencial; USD 168 000 - 227 000; Tempo integral

Resumo da oferta

Annapurna Labs (U.S.) Inc. in Seattle seeks a Senior Software Engineer to advance AWS Neuron’s AI accelerators. You will optimize deep learning models and work across compiler, runtime, framework, and hardware teams to push performance on Inferentia and Trainium.

You will design kernels, monitor system performance, and collaborate with customers to maximize efficiency of ML workloads on AWS accelerators.

Qualificações

• Bachelor’s degree required.
• 5+ years of professional software development experience.
• Knowledge of Python and/or C++ programming.

Responsabilidades

• Lead distributed inference support for PyTorch in the Neuron SDK.
• Tune models for highest performance on Trainium/Inferentia silicon.
• Collaborate across compiler, runtime, framework, and hardware teams.
• Design and implement high-performance ML kernels and features for Neuron.

Conhecimentos

Python programming

C++ programming

Performance profiling

Parallel computing

System design leadership

Formação académica

Master’s degree in Computer Science or equivalent

Bachelor’s degree

Ferramentas

PyTorch

CUDA

Triton

NCCL

Descrição da oferta de emprego

Annapurna Labs (U.S.) Inc. in Seattle seeks a Senior Software Engineer to advance AWS Neuron’s AI accelerators. You will optimize deep learning models and work across compiler, runtime, framework, and hardware teams to push performance on Inferentia and Trainium.

You will design kernels, monitor system performance, and collaborate with customers to maximize efficiency of ML workloads on AWS accelerators.