Machine Learning Engineer (Platform)

Position Overview

• Location Type: Remote
• Employment Type: Full-Time
• Salary: $140,000 - $180,000 a year

Artera is an AI startup developing medical artificial intelligence tests to personalize therapy for cancer patients. As a Machine Learning Engineer, you’ll contribute to the AI Platform team, building scalable and efficient pipelines for data processing and model training. You’ll collaborate with AI model developers, other machine learning engineers, and the platform engineering team to ensure efficient model development and deployment.

Requirements

• Experience:
  • 4+ years of industry software engineering experience
  • 3+ years of industry experience using PyTorch, TensorFlow, or JAX in Python
  • 2+ years of industry experience building with AWS, Docker, and Kubernetes
  • 1+ years of industry experience optimizing large-scale, high data-throughput, distributed machine learning training pipelines
• Desired (Bonus):
  • Experience using Terraform
  • Experience using SQLAlchemy
  • Experience using ML orchestration frameworks such as Ray, Kubeflow, Metaflow, MLFlow, Flyte, Dagster, Argo Workflow or Prefect
  • Experience deploying and maintaining infrastructure for machine learning training and production inference
  • Familiarity with TorchScript, ONNXRuntime, DeepSpeed, AWS Neuron or similar approaches to inference optimization

Responsibilities

• Build and own tools and libraries that accelerate Artera’s ability to develop, launch, and monitor AI products.
• Work with model developers to optimize GPU and CPU efficiency and data throughput of large-scale foundation models and downstream model training runs.
• Optimize Artera’s ability to store and process terabytes of digital pathology data efficiently for use in serving large-scale training regimes.
• Ensure that Artera’s observability infrastructure provides a clear picture of how to continue to optimize performance across our model landscape.

Application Instructions

• [Instructions for applying would be added here if available]

Company Information

• About Artera: Artera is an AI startup that develops medical artificial intelligence tests to personalize therapy for cancer patients.
• Mission: To personalize medical decisions for patients and physicians on a global scale.
• Compensation: Base salary is competitive and commensurate with experience, qualifications, and other factors to be discussed during the interview process. Equity is a core component of our compensation.
• Benefits: 401k matching, unlimited paid time off (PTO)

Equal Employee Opportunity

At Artera, 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.