Software Engineer

Software Development Engineer, Neuron Foundation Tools | Seattle, Washington | Remote-Friendly | $129,300 - $223,600

We're working with Annapurna Labs (U.S.) Inc. on this exciting opportunity.

Join Annapurna Labs (U.S.) Inc. as a Software Development Engineer on the Neuron Foundation Tools Team, where you'll develop and maintain high-performance monitoring and profiling tools for machine learning applications and AI accelerators. This role offers the chance to make a significant impact on optimizing AI workloads and provides a supportive environment for career growth and mentorship.

Key Responsibilities:

• Develop and maintain high-performance monitoring and profiling tools for machine learning applications and AI accelerators.

• Work on the design, development, and deployment of the Neuron Profiler and other Neuron Tools.

• Manage the full development life cycle of the Neuron Profiler/Tools toolchain, ensuring scalability, reliability, and usability.

• Collaborate with cross-functional teams to ensure the C++ compiler and runtime generate key optimization information.

• Drive innovations to support multiple frameworks like PyTorch, JAX, and XLA within the profiler.

What You'll Need:

• 3+ years of non-internship professional software development experience.

• Established background in building AI/ML and performance analysis tools.

• Experience with ML-specific profiler tools (like PyTorch Profiler or TensorFlow Profiler) is highly desirable.

• Strong motivation to achieve results.

What's On Offer:

• Inclusive Team Culture with employee-led affinity groups and ongoing learning experiences.

• Work/Life Balance with flexibility in working hours.

• Mentorship & Career Growth opportunities with a focus on knowledge sharing and project-based development.

Apply via Haystack today