Compute Software Engineer (AI)

TL;DR

Compute Software Engineer (AI): Building and optimizing GPU-first cloud infrastructure for bare-metal and virtual machine provisioning with an accent on performance, scalability, and reliability. Focus on designing distributed systems for compute resource orchestration, troubleshooting complex production issues, and streamlining the software delivery lifecycle.

Location: Must be based in or able to commute to San Francisco, San Jose, or Bellevue offices 4 days per week.

Salary: $266,000 – $395,000

Company

Lambda is a leader in AI cloud infrastructure, providing high-performance compute resources to researchers and enterprises to power superintelligence.

What you will do

• Design and maintain software for GPU/CPU compute infrastructure.
• Implement services for bare-metal and VM instancing.
• Develop distributed systems for orchestrating compute resources across various SKUs.
• Troubleshoot and debug complex issues in production and development environments.
• Participate in on-call rotations and incident ownership.
• Collaborate across engineering teams to drive technical requirements and RFCs.

Requirements

• 3+ years of experience with Go or Python in production environments.
• 3+ years of experience with bare-metal and virtualization hardware management.
• Proficiency in Linux environments and debugging at OS, hardware, and networking layers.
• Ability to independently troubleshoot complex systems.
• Must be able to work from the San Francisco, San Jose, or Bellevue office 4 days per week.

Nice to have

• Familiarity with GPU infrastructure or high-performance computing.
• Experience with Slurm or Kubernetes-based cluster management.
• Knowledge of public cloud internals like KVM, QEMU, and security.
• Experience with durable execution platforms like Temporal.

Culture & Benefits

• Generous cash and equity compensation.
• Health, dental, and vision coverage for employees and dependents.
• 401k plan with 2% company match.
• Wellness and commuter stipends.
• Flexible paid time off policy.