Senior Software Engineer, AI Infrastructure

AI2
Seattle
Workplace: OnsiteFull timeUSD 126,000 - 189,000Function: Software EngineeringExperience: 8+ yearsEducation: bachelorsSkills: ["Communication","Leadership","Mentorship","Problem-solving","Collaboration"]

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.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
AI2
AI2
2 months 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: 2 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
WebsiteLinkedIn