Akshay Nalwaya

Software Development Engineer at AWS-Annapurna Labs

Role
Software Engineer at Amazon Web Services (AWS)
Location
Seattle, WA, US
LinkedIn followers
500 followers
Information TechnologyView LinkedIn profile

About Akshay Nalwaya

As a Software Development Engineer II at Amazon Web Services (AWS), I focus on developing and scaling machine learning systems and developer tooling for the AWS Neuron training and inference stack. My contributions include compiler and runtime optimizations, debugging frameworks, and production infrastructure improvements, which have enhanced model deployment efficiency and reduced latency. With over six years of combined experience in software engineering and machine learning, I have contributed to backend and ML platform capabilities, driving automation, infrastructure efficiency, and productivity improvements. My work is rooted in leveraging skills in Python, Java, deep learning, and data science to deliver impactful solutions that streamline workflows and enhance system reliability.

Experience

  1. Software Engineer

    Amazon Web Services (AWS)

    Aug 2024 — Present · Seattle, WA, US

    I build and scale ML systems and developer tooling for AWS Neuron’s training and inference stack. My work spans compiler and runtime optimizations, configuration and debugging frameworks, and production infrastructure enhancements that reduce model bring-up and deployment friction for internal teams and customers. I have delivered significant reductions in compilation and tracing overhead through targeted metadata extraction and by eliminating redundant graph work. Improved inference latency by 10–15% via system-level optimizations, and reduced end-to-end model trace time by ~10–15% by introducing an HLO-keyed compiled-artifact caching layer. I also focus on engineering productivity and reliability by building automation and testing workflows that raise coverage and reduce manual effort, while partnering cross-functionally to ship scalable solutions for latest multi-modal architectures.

Education

  • North Carolina State University

    Master's degree, Computer Science

    2017 — 2019

  • SGSITS

    Bachelor's degree, Computer Science

    2012 — 2016

Skills

  • C
  • Programming
  • Powerpoint
  • C++
  • Java
  • Python
  • Event Planning
  • Android Development
  • Mysql
  • R
  • Microsoft Office
  • Microsoft Powerpoint

Find verified contacts for anyone on LinkedIn

Unifers gives sales teams verified emails and direct dials, enriched profiles, and outreach that lands in the inbox.

Free plan included · No credit card required

This profile is compiled from publicly available professional sources. Unifers is not affiliated with or endorsed by LinkedIn. Request removal of this profile.

Akshay Nalwaya — Software Engineer at Amazon Web Services (AWS) in Seattle, WA, US | Unifers