Software Development Engineer on the Neuron Runtime team at Annapurna Labs (AWS), building and maintaining high-performance runtime libraries, drivers, and profiler software for AWS Trainium/Inferentia AI accelerator chips. Involves full development lifecycle ownership, C++ compiler integration, and framework support for PyTorch, JAX, and XLA, plus distributed systems and AWS services experience.
Sign in to see your match