AI2

1 month ago

Senior Software Engineer, AI Infrastructure

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI

Last checked: 4 hours agoStatus: Live

Job Summary

Senior technical contributor responsible for designing and scaling the Beaker job scheduler and GPU execution runtime, bridging researchers and GPU clusters. You will drive automation, performance engineering, and distributed-systems improvements to enable world-class AI model training at scale in a non-profit research environment.

Location: Seattle

Workplace: Onsite

Employment Type: Full time

Job Function: Software Engineering

Key Responsibilities

•Full-Stack Ownership: Independently design and deliver critical systems that span the entire stack—from the Beaker job scheduler to the execution runtime.
•System Automation: Build innovative tooling and software-defined infrastructure to accelerate researcher velocity and automate cluster health management.
•Performance Optimization: Conduct root-cause analysis on complex distributed system failures and implement optimizations for distributed workloads.
•Technical Input & Ownership: Provide valuable input into the roadmap for managing large-scale HPC systems, including the deployment of compute, networking, and storage in partnership with leadership.
•Mentorship & Culture: Foster a high-performance culture by reviewing code/design docs, mentoring team members, and driving process improvements within the team.

Pay and Benefits

Salary: USD 126,000 - 189,000

Perks:Health InsuranceDentalVision401kCommuting StipendWellness StipendPaid LeavePaid HolidaysAnnual Bonus

Key Requirements

•8+ years of professional experience developing business-critical software and operating large-scale compute infrastructure. Proficiency in Go and/or Python preferred.
•Bachelor’s degree in related field; relevant advanced degree may substitute for equivalent years of technical work experience.
•Linux Expertise: Expert-level knowledge of Linux internals, and container runtimes like Docker.
•Distributed Systems Expertise: A proven track record of designing, debugging, and optimizing high-scale distributed systems and databases.
•Communication: Exceptional writing skills and the ability to drive consensus across diverse groups of researchers and engineers.

Experience:8+ yearsHPCAI infrastructureGPU clustersOpen Science

Education:Bachelor's

Skills:CommunicationLeadershipMentorshipProblem-solvingCollaboration

Languages:English

Tech Stack:GoPythonLinuxDockerNCCLInfiniBandKubernetesSlurm

Company Brief

AI2

The Allen Institute for AI (AI2) is a nonprofit research institute that builds AI systems and conducts fundamental research to contribute to scientific discovery, natural language understanding, and open-source tools for the AI community.

Industry: AI & Machine Learning

Company Size: Medium (51 to 250 employees)

Growth: Nonprofit & NGO

Funding: Grant Funded

Headquarters: Seattle, United States

Founded: 2014

Website LinkedIn

Loading job details...

Preparing the role view and application actions.

AI2