Machine Learning Performance Engineer

Artera is an AI startup that develops medical artificial intelligence tests to personalize therapy for cancer patients. We are on a mission to personalize medical decisions for patients and physicians on a global scale. As a Machine Learning Engineer at Artera, you’ll work on the AI Platform team with a focus on computational performance, collaborating closely with AI model developers, fellow machine learning engineers, and our platform engineering team to ensure efficient, large-scale training regimes and optimized model deployment in production environments.

Essential 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 for large-scale foundation models and downstream training runs.
• Optimize storage solutions for terabytes of digital pathology data for use in large-scale training regimes.
• Ensure observable infrastructure provides clarity on performance optimization across our model landscape.

Experience Requirements:

• 5+ years of industry software engineering experience.
• 3+ years using PyTorch, TensorFlow, or JAX in Python.
• 2+ years building with AWS, Docker, and Kubernetes.
• 1+ years optimizing large-scale, high data-throughput, distributed machine learning training pipelines.

Desired Experience:

• Using Terraform and SQLAlchemy.
• Experience with ML orchestration frameworks such as Ray, Kubeflow, Metaflow, MLFlow, Flyte, Dagster, Argo Workflow, or Prefect.
• Deploying and maintaining infrastructure for machine learning training and production inference.
• Familiarity with TorchScript, ONNXRuntime, DeepSpeed, AWS Neuron or similar approaches to inference optimization.

Compensation:

• Salary: $140,000 - $200,000 a year, commensurate with experience and qualifications.
• Equity is a core component of our compensation package.
• Additional benefits include 401k matching, unlimited paid time off (PTO), and more.

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 based on 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.