Senior AI Infra Engineer, AI/ML and Data Infrastructure

Company: Chan Zuckerberg Initiative
Location: Redwood City
Posted on: November 6, 2024

Job Description:

The Team
Across our work in Science, Education, and within our communities, we pair technology with grantmaking, impact investing, and collaboration to help accelerate the pace of progress toward our mission. Our Central team provides the support needed to push this work forward.
The Central team at CZI consists of our Finance, People & DEI, Real Estate, Events, Workplace, Facilities, Security, Brand & Communications, Business Systems, Central Operations, Strategic Initiatives, and Ventures teams. These teams provide strategic support and operational excellence across the board at CZI.
The Opportunity
By pairing engineers with leaders in our science and education teams, we can bring AI/ML technology to the table in new ways to help drive solutions. We are uniquely positioned to design, build, and scale software systems to help educators, scientists, and policy experts better address the myriad challenges they face. Our technology team is already helping schools bring personalized learning tools to teachers and schools across the country. We are also supporting scientists around the world as they develop a comprehensive reference atlas of all cells in the human body, and are developing the capacity to apply state-of-the-art methods in artificial intelligence and machine learning to solve important problems in the biomedical sciences.The AI/ML and Data Engineering Infrastructure organization works on building shared tools and platforms to be used across all of the Chan Zuckerberg Initiative, partnering and supporting the work of a wide range of Research Scientists, Data Scientists, AI Research Scientists, as well as a broad range of Engineers focusing on Education and Science domain problems. Members of the shared infrastructure engineering team have an impact on all of CZI's initiatives by enabling the technology solutions used by other engineering teams at CZI to scale. A person in this role will build these technology solutions and help to cultivate a culture of shared best practices and knowledge around core engineering.
What You'll Do

• 
  
• Participate in the technical design and building of efficient, stable, performant, scalable and secure AI/ML and Data infrastructure engineering solutions.
  
• Active hands-on coding working on our Deep Learning and Machine Learning models
  
• Design and implement complex systems integrating with our large scale AI/ML GPU compute infrastructure and platform, making working across multiple clouds easier and convenient for our Research Engineers, ML Engineers, and Data Scientists.
  
• Use your solid experience and skills in building containerized applications and infrastructure using Kubernetes in support of our large scale GPU Research cluster as well as working on our various heterogeneous and distributed AI/ML environments.
  
• Collaborate with other team members in the design and build of our Cloud based AI/ML platform solutions, which includes Databricks Spark, Weaviate Vector Databases, and supporting our hosted Cloud GPU Compute services running containerized PyTorch on large scale Kubernetes.
  
• Collaborate with our partners on data management solutions in our heterogeneous collection of complex datasets.
  
• Help build tooling that makes optimal use of our shared infrastructure in empowering our AI/ML efforts with world class GPU Compute Cluster and other compute environments such as our AWS based services.
  
  What You'll Bring
  
  • 
    
  • BS or MS degree in Computer Science or a related technical discipline or equivalent experience
    
  • 5+ years of relevant coding experience
    
  • 3+ years of systems Architecture and Design experience, with a broad range of experience across Data, AI/ML, Core Infrastructure, and Security Engineering
    
  • Scaling containerized applications on Kubernetes or Mesos, including expertise with creating custom containers using secure AMIs and continuous deployment systems that integrate with Kubernetes or Mesos. (Kubernetes preferred)
    
  • Proficiency with Amazon Web Services (AWS), Google Cloud Platform (GCP), or Microsoft Azure, and experience with On-Prem and Colocation Service hosting environments
    
  • Proven coding ability with a systems language such as Rust,C/ C++, C#, Go, Java, or Scala
    
  • Shown ability with a scripting language such as Python, PHP, or Ruby
    
  • AI/ML Platform Operations experience in an environment with challenging data and systems platform challenges - including large scale Kafka and Spark deployments (or their coralaries such as Pulsar, Flink, and/or Ray) as well as Workflow scheduling tools such as Apache Airflow, Dagster, or Apache Beam
    
  • MLOps experience working with medium to large scale GPU clusters in Kubernetes (Kubeflow), HPC environments, or large scale Cloud based ML deployments
    
  • Working knowledge of Nvidia CUDA and AI/ML custom libraries.
    
  • Knowledge of Linux systems optimization and administration
    
  • Understanding of Data Engineering, Data Governance, Data Infrastructure, and AI/ML execution platforms.
    
  • PyTorch, Karas, or Tensorflow experience a strong nice to have
    
  • HPC with and Slurm experience a strong nice to have
    
    Compensation
    The Redwood City, CA base pay range for this role is $190,000 - $285,000. New hires are typically hired into the lower portion of the range, enabling employee growth in the range over time. Actual placement in range is based on job-related skills and experience, as evaluated throughout the interview process. Pay ranges outside Redwood City are adjusted based on cost of labor in each respective geographical market. Your recruiter can share more about the specific pay range for your location during the hiring process.
    Benefits for the Whole You
    We're thankful to have an incredible team behind our work. To honor their commitment, we offer a wide range of benefits to support the people who make all we do possible.
    
    • 
      
    • CZI provides a generous employer match on employee 401(k) contributions to support planning for the future.
      
    • Annual benefit for employees that can be used most meaningfully for them and their families, such as housing, student loan repayment, childcare, commuter costs, or other life needs.
      
    • CZI Life of Service Gifts are awarded to employees to "live the mission" and support the causes closest to them.
      
    • Paid time off to volunteer at an organization of your choice.
      
    • Funding for select family-forming benefits.
      
    • Relocation support for employees who need assistance moving to the Bay Area
      
    • !
      
      Commitment to Diversity
      We believe that the strongest teams and best thinking are defined by the diversity of voices at the table. We are committed to fair treatment and equal access to opportunity for all CZI team members and to maintaining a workplace where everyone feels welcomed, respected, supported, and valued.about our diversity, equity, and inclusion efforts.
      If you're interested in a role but your previous experience doesn't perfectly align with each qualification in the job description, we still encourage you to apply as you may be the perfect fit for this or another role.
      Explore our,, andat.
      #LI-Remote
      #J-18808-Ljbffr

Keywords: Chan Zuckerberg Initiative, Alameda , Senior AI Infra Engineer, AI/ML and Data Infrastructure, Accounting, Auditing , Redwood City, California