cuspai - Platform Engineer
Requirements
• Experience using our tech stack (Flyte, Kapitan, Pants) or similar tools • Experience with HPC environments and job schedulers (Slurm, PBS) • Knowledge of ML serving infrastructure (Triton, TorchServe, KServe) • Familiarity with scientific computing workflows and data management • Experience with multi-tenant ML platforms • Background in supporting research teams (understanding their unique needs vs. production ML)
Responsibilities
• Build the ML Infrastructure Platform: Design and implement a cloud-native platform on Kubernetes that enables researchers to easily train, evaluate, and deploy models without worrying about infrastructure complexity. • Own the MLOps Stack: Implement and maintain CI/CD pipelines, model registries, experiment tracking systems, and deployment automation - the full lifecycle of ML operations. • Scale Distributed ML Model Training: Build infrastructure to support distributed training across many GPUs, implementing solutions for data pipelines, checkpointing, and resource optimisation. • Platform Reliability: Ensure 99.9% uptime for our ML platform through monitoring, alerting, and automated recovery systems. Be the guardian of infrastructure stability. • Configuration Management: Use Kapitan to manage complex multi-environment Kubernetes configurations and ensure consistent deployments across dev, staging, and production. • Cost Optimisation: Implement resource management strategies to optimise cloud spending and detect inefficient usage while maximising computational throughput for our research teams. • Developer Experience: Create tools and abstractions that make it simple for researchers to go from experiment to production without deep infrastructure knowledge. • Interdisciplinary Collaboration: Partner closely with ML Researchers, Chemists, Materials Scientists and Software Engineers to understand their needs, and to build the infrastructure for groundbreaking projects
Benefits
• A competitive salary: We value and reward impact and growth • Equity in CuspAI: You have a stake in the success of the company • Time off to stay fresh: 28 days holiday (DE, NL, UK) or 21 days holiday (JP, SG, US), in addition to local public holidays • ‘Gold Standard’ parental leave: 26 weeks (primary caregiver) and 12 weeks (secondary caregiver) at full pay - we look after you and your family while we work on the most important materials discovery problems together • Professional development budget: We invest in your career development so you can stay up to date with the latest industry knowledge or add to your skills to increase impact and growth • Solve meaningful problems: See how your work has a direct impact on advancing materials science and solving sustainability and climate-related problems through the creation and application of bleeding-edge SOTA technology and revolutionary techniques • True interdisciplinary teamwork: Be part of a deeply collaborative environment bridging AI research, computational chemistry, and experimental science - work with world-class researchers and engineers who enjoy sharing knowledge and supporting each other • CuspAI is an equal opportunities employer committed to building a diverse and inclusive workplace. We do not discriminate on the basis of sex, race, religion or belief, ethnic or national origin, disability, age, citizenship, marital, domestic or civil partnership status, sexual orientation, gender identity, pregnancy or related condition (including breastfeeding), veteran status, or any other basis protected by applicable law. • We actively encourage applications from all backgrounds and value the unique perspectives and contributions that diversity brings to our team. • Please let us know if you require any specific adjustments during or after the interview process. We will do everything we can within reason to accommodate.
Apply in one click
Upload My Resume
Drop here or click to browse · Tap to choose · PDF, DOCX, DOC, RTF, TXT