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 engine

Requirements: • Bachelor's degree in Computer Science, Engineering, or related field • 3+ years of skilled software development experience • 2+ years of design or architecture experience • Proficient in at least one software programming language • Experience with PyTorch or JAX preferred

Deliver impactful machine learning optimization tools as a part of the Neuron Explorer team at Annapurna Labs.

📌 Software Engineer for Neuron Explorer (Winnipeg)
🏢 Socket.dev
📍 Winnipeg