Join Annapurna Labs as a Software Development Engineer on the AWS Neuron team in Toronto. Design and build tools that optimize machine learning workloads to enhance performance on Trainium and Inferentia.In this role, you will take charge of the Neuron Explorer, focusing on performance data analysis and visualization. You will design end-to-end features, from backend data collection to interactive user interfaces, ensuring reliability and ease of use. Collaborate with cross-functional teams to effectively surface performance metrics, enabling customers to maximize their AI workloads.Key Responsibilities:
Design performance data collection and analysis pipelines
Develop interactive visualizations for profiling data
Own features through the entire software development lifecycle




Collaborate with compiler and runtime teams for optimizing workloads
Integrate generative AI capabilities into recommendation engineRequirements:
Bachelor's degree in Computer Science, Engineering, or related field
3+ years of qualified software development experience
2+ years of design or architecture experience
Proficient in at least one software programming language
Experience with PyTorch or JAX preferredDeliver impactful machine learning optimization tools as a part of the Neuron Explorer team at Annapurna Labs.#J-18808-Ljbffr

📌 Software Engineer For Neuron Explorer Toronto (Canada)
🏢 Socket.dev
📍 Canada