Software Development Engineer, ML Systems, Annapurna Labs
Core
Building AI agents, tools, and models to simplify and accelerate customer adoption of AWS Neuron (software stack for Trainium/Inferentia ML chips) and optimizing ML workloads on Neuron.
Role type
Senior IC software development engineer (ML systems & GenAI)
Builds
AI agents, optimization tools, and software solutions for AWS Neuron adoption
Domain
Cloud infrastructure, Machine Learning hardware, Generative AI
Deliverable
production ML models | product features
Required skills
ML compilers, GenAI model architecture, neural network optimization, applied math, software architecture, distributed inference and training
Preferred skills
ML kernel authoring, open-source optimization, state-of-the-art ML technology knowledge
Technologies
AWS Neuron, Trainium, Inferentia, GenAI frameworks
Responsibilities
Research implementations for best customer experience, improve porting and optimization time for ML workloads on Neuron, design and deploy innovative software solutions, build high-availability products, contribute to patents
Seniority
Senior, hands-on IC
