Compute Software Engineer

Responsibilities

• Provide technical leadership and application development expertise across multidisciplinary AI and advanced computing efforts.
• Own the software architecture for multiple codebases and select optimal tools to enable new scientific capabilities.
• Implement recently published algorithms and novel data-processing techniques within existing applications.
• Instrument research software for performance optimization and analysis, specifically for large-scale, distributed workflows.
• Construct scalable AI services using a combination of open-source packages and proprietary APIs across hybrid on-prem and cloud resources.
• Troubleshoot and evaluate complex software packages for numerical accuracy and hardware platform adoption.
• Mentor existing programs and collaborate with subject matter experts to implement new software features.

Requirements

• Bachelor's degree in Computer Science, Mathematics, or a related field.
• 5+ years of professional software development experience producing high-quality applications.
• Proficiency with modern machine-learning frameworks such as JAX, PyTorch, or TensorFlow.
• Exposure to CPU/GPU/TPU acceleration (e.g., CUDA, ROCm, Cloud TPU).
• Experience with containerized Linux workloads, including GPU-accelerated configurations.
• Knowledge of computer networking and distributed computing.
• Ability to design, train, deploy, and maintain state-of-the-art AI models at production scale.
• Ability to travel domestically and internationally as needed (up to 30%).

Preferred Qualifications

• Expert-level experience in software and hardware frameworks driving modern AI.
• Expertise in AI model optimization and hardware-accelerated frameworks.
• Experience evaluating scalability, data requirements, and compute needs throughout the software design lifecycle.
• Ability to translate technical concepts across multiple scientific disciplines.
• Experience working with academic, government, or philanthropic research institutions.

About the Company

Schmidt Sciences is a nonprofit organization founded by Eric and Wendy Schmidt to accelerate scientific knowledge and breakthroughs. Through the AI and Advanced Computing Institute, the organization supports research in AI for science, the science of AI, and beneficial AI to create the conditions for AI-enabled discovery to achieve its promise.