Software Engineer - Compute Platform

About Us: Keystone AI is an AI startup that develops medical artificial intelligence tests to personalize therapy for cancer patients. Keystone AI is on a mission to personalize medical decisions for patients and physicians on a global scale. Compute Infrastructure is at the core of Keystone AI 's products - using AI to interpret gigapixel-scale images of tissue biopsies to make predictions about cancer therapy, as well as rapidly developing new AI models for global deployment. As a Software Engineer on the Platform Engineering team, you'll work alongside fellow Platform Engineers to develop tools and services that improve developer experience, security, and productivity through self-service capabilities and automated infrastructure operations. You'll closely collaborate with AI scientists, ML engineers, and software developers to design, build, and maintain the infrastructure that powers everything Keystone AI does at scale. Essential Responsibilities: Design, build, and maintain compute infrastructure programmatically (AWS Kubernetes/EKS, AWS ECS, Lambda, and EC2) that powers Keystone AI 's AI products at scale. Work closely with stakeholders to define and refine the platform's architecture, ensuring scalability, observability, reliability, and performance. Build out core infrastructure, tooling, and software development processes. Work closely with machine learning engineers to optimize training and inference workflows with efficiency and cost in mind. Contribute to a range of platform engineering projects, from one-off solutions to long-term systems. Contribute to cloud infrastructure security tools and services. Experience Requirements: 2+ years managing containerized infrastructure at scale with a strong security mindset. 3+ years building services and tools using Python with a software engineering mindset. 2+ years building infrastructure automation solutions using Infrastructure-as-Code (e.g., Terraform, AWS CDK). Experience with AWS storage services: S3, EFS, FSx. Experience with CI/CD pipelines. Demonstrated integrity and consideration for appropriate data governance when working with sensitive, confidential data. Desired: Experience with infrastructure observability and monitoring tools (e.g., Grafana, Prometheus, Datadog). Experience supporting ML/AI workloads — GPU instance management, training cluster optimization, batch inference pipelines. Familiarity with cost optimization strategies for cloud compute at scale. Experience with secrets management and cloud security tooling (e.g., AWS IAM, Vault, KMS). Contributions to internal developer tooling or platform libraries consumed by cross-functional teams. Here are few posts from our teammates, partners and customer voices to highlight the work we do: Keystone AI Shapes the Future of Cancer Treatment Using Machine Learning on AWS How Keystone AI AI test allowed Bruno to avoid hormone therapy Startups are using AI to predict responses to Cancer Drugs Keystone AI AI validates its Prognostic Model in Advanced Prostate Cancer How Keystone AI Enhances Prostate Cancer Diagnostics Using AWS Work Authorization Requirement: This is a remote role open to candidates who are currently authorized to work either in the United States or in Canada without the need for current or future employment-based visa sponsorship. Keystone AI does not sponsor visas for this position. Eligible candidates may include: Individuals authorized to work in the United States on a permanent basis(eg. U.S. citizens, permanent residents), or Individuals authorized to work in Canada (e.g. Canadian citizens or Canadian Permanent residents) Visa Transfers (if needed) Equal Employee Opportunity : At Keystone AI , we value bringing together individuals from diverse backgrounds to develop new and innovative solutions for patients and physicians. As an equal opportunity employer, we do not discriminate on the basis of race, color, religion, national origin, age, sex (including pregnancy), physical or mental disability, medical condition, genetic information gender identity or expression, sexual orientation, marital status, protected veteran status, or any other legally protected characteristic. Originally posted on Himalayas