ArteraAI · 1 day ago
Machine Learning Engineer (AI Platform)
United States
Full-time
Remote
Mid, Senior Level
$140K/yr - $180K/yr
4+ years expArteraAI is an AI startup focused on developing medical artificial intelligence tests to personalize therapy 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 while collaborating with model developers to optimize performance and efficiency.
Artificial Intelligence (AI)BiotechnologyHealth CareLife SciencePrecision Medicine
No H1B
Responsibilities
Build and evolve the core libraries used by AI scientists to develop, launch, and monitor AI products
Help manage Artera’s ML compute infrastructure including scaling up Artera’s Foundation Model development by developing distributed training infrastructure and developer libraries
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
Required
4+ years of industry software engineering experience
2+ years of industry experience in using ML orchestration frameworks such as Flyte, Ray, Kubeflow, Metaflow, MLFlow, Dagster, Argo Workflow or Prefect
2+ 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
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
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
Our AI can prognosticate patient outcomes & personalize treatment in localized prostate cancer.
51-200 employees
Funding
Current Stage
Growth StageTotal Funding
$110M2024-02-01Series Unknown· $20M
2023-03-21Series Unknown· $90M
Leadership Team
Peter Wood
Strategic Partnerships
Recent News
2025-12-09
vcnewsdaily.com
2025-12-04
Company data provided by crunchbase