Software Development Engineer, ML Systems, Annapurna Labs

Role responsibilities

Build AI agents, tools, and models to simplify and accelerate the adoption of AWS Neuron software for Trainium ML silicon. Collaborate cross-functionally to research, design, and deploy innovative software solutions that optimize ML workloads.

Requirements

Requires 3+ years of professional software development experience and 2+ years of system design or architecture experience. Candidates should have expertise in computer science cores and experience in ML compilers, GenAI architecture, or neural network optimization.

Key skills

Software Development, System Architecture, Machine Learning, Generative AI, ML Compilers, Neural Network Optimization, Object-Oriented Design, Data Structures, Performance Analysis, AI Agents, Model Training, Distributed Inference, ML Kernel Optimization, Applied Math, Source Control Management, Testing

Keywords

AWS Neuron, Trainium, Inferentia, Generative AI, ML Systems, Silicon Engineering, AWS Nitro, Graviton, ML Compiler, Neural Networks, Distributed Training, AI Agents, Software Architecture, Performance Analysis, Cloud Infrastructure, AWS, Annapurna Labs, Object-Oriented Design, Model Optimization, Kernel Authoring