200+ applicants

Company

ArteraAI · 2 days ago

Machine Learning Performance Engineer

United States

Full-time

Remote

Senior Level

$140K/yr - $200K/yr

5+ years exp

Maximize your interview chances

Artificial Intelligence (AI)Biotechnology

H1B Sponsor Likely

Insider Connection @ArteraAI

Discover valuable connections within the company who might provide insights and potential referrals.
Get 3x more responses when you reach out via email instead of LinkedIn.

Responsibilities

Build and own tools and libraries that accelerate Artera’s ability 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 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

Find out how your skills align with this job's requirements. If anything seems off, you can easily click on the tags to select or unselect skills to reflect your actual expertise.

PythonPyTorchTensorFlowAWSDockerKubernetesJAXTerraformSqlAlchemyML orchestration frameworksKubeflowMetaflowMLFlowFlyteDagsterArgo WorkflowPrefectInfrastructure maintenanceInternal tools development

Required

5+ years of industry software engineering experience

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

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

Glassdoor

3.4

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

Founded in 2021

Los Altos, California, USA

51-200 employees

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

2023 (7)

2022 (5)

2021 (3)