# Machine Learning Engineer (AI Platform Lead)

**Company:** [Artera](https://hotfix.jobs/companies/artera)
**Location:** Remote
**Role:** ML Engineering
**Salary:** $140k – $180k/yr
**Experience:** 5+ years
**Skills:** PyTorch, TensorFlow, JAX, Python, AWS, Docker, Kubernetes, Flyte, Ray, Kubeflow, Metaflow, MLflow, Dagster, Argo Workflow, Prefect
**Posted:** 2026-01-22

> Build and scale ML compute infrastructure and distributed training pipelines for foundation models. Optimize GPU/CPU efficiency and data throughput for large-scale model training and inference.

## Job Description

## Essential Responsibilities
- 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.

## Experience Requirements
- 5+ years of industry software engineering experience
- 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

## Desired
- Experience in using ML orchestration frameworks such as Flyte, Ray, Kubeflow, Metaflow, MLFlow, Dagster, Argo Workflow or Prefect
- 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

## Similar roles

- [Senior Data Scientist - International eKYC, Identity Graph](https://hotfix.jobs/jobs/d287b4f4-bba2-407f-bdd8-62d5d9b67b5c) - Socure - Remote - $140k – $170k/yr
- [Senior Data Scientist - Big Data R&D, Identity Graph & KYC](https://hotfix.jobs/jobs/7c47bd1f-5c10-4996-b613-5a2bf72af886) - Socure - Remote - $140k – $170k/yr
- [Senior Software Engineer, Agents](https://hotfix.jobs/jobs/1d321f27-96d0-43db-9da1-89d2e884709b) - Commure - Mountain View, CA - $140k – $230k/yr
- [Senior AI Engineer](https://hotfix.jobs/jobs/006c53e2-ed99-4ec4-afa9-f7edf007af5e) - GitLab - Remote - $139k – $218k/yr
- [Senior Machine Learning Engineer](https://hotfix.jobs/jobs/2ea60d0a-0e5f-4d54-ba8f-47ffc5a1f1b5) - Mozilla - Remote - $139k – $218k/yr

**Apply:** https://hotfix.jobs/jobs/9c21648c-b474-479d-88dd-34060c80b3be
**Canonical:** https://hotfix.jobs/jobs/9c21648c-b474-479d-88dd-34060c80b3be