Software Development Engineer • Mid-Level

Who you are

You have a strong background in software development, particularly in building AI and machine learning tools. Your experience includes working with performance analysis tools and optimizing machine learning workloads across various hardware platforms. You are proficient in C++ and have a solid understanding of AWS services, which allows you to effectively contribute to the development of high-performance tools. You are familiar with ML frameworks such as PyTorch and TensorFlow, and you understand the intricacies of profiling and optimizing AI workloads. Your collaborative spirit enables you to work well with cross-functional teams, ensuring that the tools you develop meet the needs of both internal and external customers.

You have experience managing the full development life cycle of software tools, ensuring they are scalable, reliable, and user-friendly. You are driven by innovation and are eager to explore new ways to enhance the capabilities of profiling tools. Your customer-facing experience has equipped you with the skills to understand user needs and translate them into effective solutions. You are detail-oriented and committed to delivering high-quality software that meets performance benchmarks.

What you'll do

As a Software Development Engineer on the Neuron Foundation Tools Team, you will be responsible for designing, developing, and deploying the Neuron Profiler and other related tools. Your work will involve collaborating with engineers to create high-performance monitoring and profiling tools that provide deep insights into performance bottlenecks and system behavior. You will focus on improving the performance of machine learning kernels and frameworks, ensuring that the tools you develop are effective across various hardware platforms, including Trainium and Inferentia devices.

You will manage the full development life cycle of the Neuron Profiler toolchain, ensuring that it is scalable and reliable. Your role will require you to drive innovations that allow the profiler to support multiple frameworks, such as PyTorch, JAX, and XLA. You will work closely with cross-functional teams to ensure that the C++ compiler and runtime generate key information that helps customers optimize the performance of their custom hardware. Your contributions will play a crucial role in enhancing the user experience and effectiveness of machine learning applications.

What we offer

At Amazon, you will be part of a dynamic team that is at the forefront of machine learning technology. We offer a competitive salary range of $143,700.00 - $194,400.00 USD annually, along with comprehensive benefits that include health insurance, retirement plans, and paid time off. You will have the opportunity to work on cutting-edge technology that impacts the future of AI and machine learning. We encourage you to apply even if your experience doesn't match every requirement, as we value diverse perspectives and backgrounds.