Senior ML Engineer: Scalable AI Platform (Remote, Equity)

Artera; United States; Remote; USD 140,000 - 180,000; Full time

Benefits offered by this job

401(k) matching

Unlimited paid time off

Equity component

Job summary

Artera is seeking a Machine Learning Engineer to develop scalable pipelines for model training and evaluation, collaborate with AI teams, and optimize model performance.

The ideal candidate will have over 5 years of software engineering experience and strong expertise in PyTorch or TensorFlow. This remote role is open to candidates authorized to work in the United States or Canada and offers a competitive salary between $140,000 - $180,000, along with equity and benefits.

Qualifications

• 5+ years of industry software engineering experience.
• 4+ years of experience using PyTorch, TensorFlow, or JAX in Python.
• 1+ year optimizing large-scale, high data-throughput ML training pipelines.

Responsibilities

• Build and evolve core libraries for AI product development.
• Optimize GPU and CPU efficiency for large-scale ML models.
• Ensure observability infrastructure optimizes model performance.

Skills

Industry software engineering

PyTorch

TensorFlow

JAX

AWS

Docker

Kubernetes

Tools

ML orchestration frameworks

Terraform

SQLAlchemy

Job description

Artera is seeking a Machine Learning Engineer to develop scalable pipelines for model training and evaluation, collaborate with AI teams, and optimize model performance.

The ideal candidate will have over 5 years of software engineering experience and strong expertise in PyTorch or TensorFlow. This remote role is open to candidates authorized to work in the United States or Canada and offers a competitive salary between $140,000 - $180,000, along with equity and benefits.