Software Engineering Manager, Neuron Tools

We are committed to fostering an inclusive team culture at AWS, embracing our diverse backgrounds and perspectives. Our organization boasts ten employee-led affinity groups impacting 40,000 employees across more than 190 chapters globally. We prioritize work-life balance, valuing the flow between personal and professional spheres to promote overall happiness and fulfillment. Flexibility in working hours is encouraged to help maintain this balance. Our dedication to mentorship and career growth means we support new team members through shared knowledge and guidance. We aim to assign projects that will help each team member develop into a well-rounded professional, enabling them to tackle more complex tasks in the future. Preferred Qualifications: - A minimum of 2 years leading teams in Machine Learning development, including building and training large models using PyTorch and/or TensorFlow on distributed systems. - Experience with Linux distributions such as Ubuntu or CentOS, kernel development, and tools like perf and gdb. - Knowledge of performance profiling, tracing, and analysis for AI training/inference applications. - Experience with large-scale, distributed AI training/inference applications, including libfabric, MPI, slurm, and EKS. - Familiarity with fleet monitoring, debugging, and ensuring reliability. - Understanding AI-powered optimization suggestions for profiling is advantageous. We celebrate our commitment to equal opportunity and inclusivity, ensuring that all Amazonians can excel in their roles while delivering exceptional results for our customers.