Machine Learning Engineer (Platform) jobs in United States
cer-icon
Apply on Employer Site
company-logo

ArteraAI · 1 month ago

Machine Learning Engineer (Platform)

positionUnited States
timeFull-time
remoteRemote
seniorityMid, Senior Level
money$140K/yr - $220K/yr
date4+ years exp

Artera is an AI startup focused on developing medical artificial intelligence tests for cancer patients. As a Machine Learning Engineer, you will work on the AI Platform team to establish scalable data processing and model training pipelines, ensuring efficient model deployment and performance optimization.

Artificial Intelligence (AI)BiotechnologyHealth CareLife SciencePrecision Medicine
check
H1B Sponsor Likelynote

Responsibilities

Evolve and manage Artera’s ML compute infrastructure
Build and evolve the core libraries used by AI scientists to develop, launch, and monitor AI products
Architect our model development and inference pipelines to accelerate model development speed and developer efficiency
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 the 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

Qualification

ML orchestration frameworksPyTorchTensorFlowAWSDockerKubernetesLarge-scale training pipelinesTerraformSqlAlchemyMulti-node trainingMulti-gpu trainingInfrastructure maintenanceInference optimization

Required

4+ years of industry software engineering experience
3+ years of industry experience in using ML orchestration frameworks such as Flyte, Ray, Kubeflow, Metaflow, MLFlow, Dagster, Argo Workflow or Prefect
3+ years of industry experience using one of 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

Preferred

Experience using Terraform, SqlAlchemy
Experience in multi-node and multi-gpu training
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

Benefits

Equity is a core component of our compensation
401k matching
Unlimited paid time off (PTO)

Company

ArteraAI

twittertwittertwitter
Glassdoor3.4
company-logo
Our AI can prognosticate patient outcomes & personalize treatment in localized prostate cancer.
dateFounded in 2021
location
Los Altos, California, USA
people-size51-200 employees
web-link
https://artera.ai

H1B Sponsorship

ArteraAI has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (3)
2024 (2)
2023 (7)
2022 (7)
2021 (1)

Funding

Current Stage
Growth Stage
Total Funding
$110M
2024-02-01Series Unknown· $20M
2023-03-21Series Unknown· $90M

Leadership Team

leader-logo
Peter Wood
Strategic Partnerships
linkedin

Recent News

Business Wire
Artera Debuts ArteraAI Prostate Test (Post‑RP), Expanding its Oncology Portfolio
2025-12-09
vcnewsdaily.com
Artera Secures $65M Growth Investment
2025-12-04
Business Wire
Artera Unveils Data Demonstrating Prognostic and Predictive Utility in Breast Cancer at SABCS 2025
2025-11-25
Company data provided by crunchbase