ML Systems Software Development Engineer Intern, Annapurna Labs - 2027
Core
Bring up and optimize state-of-the-art machine learning models for peak performance on AWS Trainium silicon and build profiling tools to identify hardware bottlenecks.
Role type
ML Systems Software Development Engineer Intern
Builds
Production ML models and profiling/debugging tools for AWS Trainium accelerators
Domain
Cloud infrastructure / Custom silicon / Machine Learning Systems
Deliverable
production ML models | product features
Required skills
Python, C, C++, performance engineering, kernel development, parallel programming, data structures and algorithms, ML frameworks (PyTorch, JAX), compiler/ML systems technologies (LLVM, MLIR, XLA, TVM)
Preferred skills
GPU/ML accelerator kernel optimization, performance analysis tooling, full stack development (TypeScript, React, Go), agentic workflows
Technologies
AWS Trainium, PyTorch, JAX, LLVM, MLIR, XLA, TVM, TypeScript, React, Go
Responsibilities
Write and tune kernels for ML models, build benchmark and measurement infrastructure, develop low-level performance data collection and analysis engines, create interactive visualizations for debugging
Seniority
Intern
