As a Software Development Engineer for the Neuron Foundation Tools Team, you will develop and maintain high-performance monitoring and profiling tools for machine learning applications and AI accelerators. Your work will focus on the design, development, and deployment of the Neuron Profiler and other tools that optimize AI workloads. You will collaborate with cross-functional teams to enhance the performance of ML Kernels and Frameworks. The role emphasizes innovation and customer engagement to support multiple frameworks.

• 3+ years of non-internship professional software development experience
• 2+ years of non-internship design or architecture experience of new and existing systems
• Experience programming with at least one software programming language
• 3+ years of full software development life cycle experience, including coding standards and testing
• Bachelor's degree in computer science or equivalent

• Develop and maintain high-performance monitoring and profiling tools for machine learning applications.
• Manage the full development life cycle of the Neuron Profiler/Tools toolchain.
• Collaborate with cross-functional teams to ensure effective performance optimization.
• Drive innovations to support multiple frameworks such as PyTorch, JAX, and XLA.
• Provide deep insights into performance bottlenecks and system behavior for internal and external customers.

• Employees at Amazon are often offered comprehensive health benefits—including multiple medical plan options (no pre-existing condition exclusions, 100% covered in-network preventive care), dental and vision plans, a 24/7 medical advice line from day one, expert second-opinion services, and broad mental-health support with several free counseling sessions (including pediatric). Financial wellness typically includes a 401(k) with company match (up to 2%), Restricted Stock Units (equity), FSAs, an emergency savings program, product and partner discounts, and even college-savings and home-purchase programs. Overall, the package is designed to support employees and their families’ health, finances, and day-to-day life.