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

ArteraAI · 11 hours ago

Machine Learning Engineer (AI Platform Lead)

ArteraAI 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 establishing scalable and efficient pipelines for data processing and model training while collaborating closely with AI model developers and the platform engineering team.

Artificial Intelligence (AI)BiotechnologyHealth CareLife SciencePrecision Medicine
badNo H1Bnote

Responsibilities

Develop the long term vision and roadmap for Artera’s AI platform that will allow the company to continue to scale in terms of both increased inference volume and development workloads
Accountable for Artera’s ML compute infrastructure including scaling up Artera’s Foundation Model development by developing distributed training infrastructure and developer libraries
Build and evolve the core libraries used by AI scientists 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 serve 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 frameworksPyTorchTensorFlowAWSDockerKubernetesDistributed trainingTerraformSqlAlchemyMulti-node trainingMulti-gpu trainingInference optimization

Required

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

company-logo
Our AI can prognosticate patient outcomes & personalize treatment in localized prostate cancer.

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
Company data provided by crunchbase