Senior Principal Software Engineer, AI Infra Compute

Job Description

Our team is the GPU Availability and Monitoring team in the Compute Org. we are responsible for designing and developing architectural changes for GPU delivery, health monitoring, triage automation, and diagnostic services. These are essential for running distributed AI/ML/HPC workloads across thousands of GPUs, leveraging technologies like RoCE and Infiniband.

We are looking for a highly skilled and motivated distributed systems engineer who can architect solutions to scale and optimize Monitoring and Repair solutions for AI infrastructure components like GPU control plane and GPU data plane that provide computing resources to customer AI workloads. You will provide technical leadership to the team and bring clarity to ambiguous problems and come up with innovative solutions.

You will collaborate with cross-functional teams to enhance our AI infrastructure to deliver exceptional customer experience and peak performance.

• Architect solutions to scale and optimize Monitoring and Repair for components like GPU, CPU, Network, Storage with the goal to optimize customer experience and customer workload performance on our AI infrastructure.
• Develop “best-in-class” AI compute infrastructure for our customers by ensuring that the services and the components are well-defined and modularized, secure, reliable, diagnosable, actively monitored, compliant and reusable.
• Collaborate with cross-functional teams, including development, operations, and product management, to understand their requirements and design respective solutions.
• Optimize and improve the software development process in the organization to improve developer efficiency.
• Mentor junior developers and drive modern software engineering practices like leveraging data/telemetry to make decisions, well-defined interfaces across components, design reviews, coding standards, code reviews, and comprehensive coverage from unit test, integration test and active production monitoring.
• Develop benchmark metrics and automation to drive and track performance and reliability across customer workload and correlate with lower infrastructure stack.
• Stay updated with industry trends, emerging technologies, and best practices in distributed systems and AI infrastructure management.

• BS (or equivalent experience) in Computer Science, Engineering, or related field.
• 10 years of experience in software development with programming languages including, but not limited to, C, C , C#, Java, Go, Rust.
• 5 years of experience designing and developing large-scale distributed systems, services and infrastructure.
• 3 years of experience providing technical leadership and clarity to cross-functional teams and projects while collaborating across stake holders.
• Systematic problem-solving approach, strong communication skills, a sense of ownership, and drive.
• Ability to adapt to a fast-paced, dynamic environment and manage multiple tasks and priorities effectively.
• Development methodologies:
  Agile Principles
• Data management: data modeling, data warehousing, data governance
• Cloud infrastructure: OCI, AWS, Azure, Google Cloud Platform (Google Cloud Platform)
• Operating Systems:
  Linux, MacOS
• Scripting languages:
  Bash, Perl, Ruby
• Familiarity with containerization technologies such as Docker
• API design and development: RESTful APIs, API gateways, API security
• Familiarity with API documentation tools such as Swagger/OpenAPI
• Experience with AI-powered tools and platforms: chatbots, virtual assistants, predictive analytics

• Experience in managing cloud infrastructure with hundreds of thousands of servers.
• Experience in containerization technologies such as Docker and Kubernetes.
• Experience in scheduling high-performance workloads on Kubernetes or Slurm.

Disclaimer:

Certain US customer or client-facing roles may be required to comply with applicable requirements, such as immunization and occupational health mandates.

Range and benefit information provided in this posting are specific to the stated locations only

US:
Hiring Range in USD from: $96,800 to $251,600 per annum. May be eligible for bonus, equity, and compensation…