Senior Software Development Engineer, AWS Neuron Frameworks

AWS Neuron is seeking a Senior Software Development Engineer to join their Frameworks team, focusing on the complete software stack for AWS Inferentia and Trainium cloud-scale machine learning accelerators. This role combines deep technical expertise in ML frameworks with strategic influence in the open-source community.

The position involves developing and enhancing support for PyTorch and JAX frameworks within AWS Neuron, working closely with the open source ecosystem. You'll be responsible for optimizing frameworks for AWS's accelerator architectures and collaborating with teams across AWS Neuron including compiler, training, and inference optimization.

As a successful candidate, you'll need strong experience in Machine Learning infrastructure and/or ML Frameworks, proven ability to work with open source communities, and excellent technical skills. Experience with technologies like XLA, vLLM, or Hugging Face transformers is highly valued.

The role offers significant opportunity for impact, working within AWS Utility Computing (UC) which provides foundational services like S3 and EC2. You'll be part of shaping the future of framework support on accelerators and enabling key features for AWS Neuron.

Amazon offers a comprehensive benefits package including medical coverage, 401k, and emphasizes work-life harmony. The company values diverse experiences and provides extensive opportunities for mentorship and career growth through their inclusive team culture. The position is based in Cupertino, CA, and offers competitive compensation ranging from $151,300 to $261,500 per year, depending on location and experience.

This is an excellent opportunity for a senior engineer passionate about machine learning infrastructure who wants to make a significant impact on the future of cloud computing while working with cutting-edge technology in a collaborative, innovative environment.