Machine Learning Engineer (Platform) - Remote

Overview

As a Machine Learning Engineer at Artera, you’ll work on the AI Platform team with a focus on establishing scalable and efficient pipelines for data processing and model training.

In Short

• 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.
• Optimize Artera’s ability to store and process terabytes of digital pathology data.
• Ensure observability infrastructure provides a clear picture of performance optimization.
• 4+ years of industry software engineering experience required.
• 3+ years of experience using PyTorch, TensorFlow, or JAX in Python.
• 2+ years of experience with AWS, Docker, and Kubernetes.
• 1+ years of experience optimizing large-scale, high data-throughput machine learning training pipelines.
• Desired experience includes Terraform, SqlAlchemy, and ML orchestration frameworks.
• Familiarity with TorchScript, ONNXRuntime, and AWS Neuron is a plus.

Requirements

• 4+ years of industry software engineering experience.
• 3+ years of experience using one of PyTorch, TensorFlow, or JAX in Python.
• 2+ years of experience building with AWS, Docker, and Kubernetes.
• 1+ years of experience optimizing large-scale, distributed machine learning training pipelines.
• Experience using Terraform, SqlAlchemy is desired.
• Experience with ML orchestration frameworks such as Ray, Kubeflow, etc.
• Experience deploying and maintaining ML infrastructure.
• Familiarity with inference optimization approaches is a plus.

Benefits

• Opportunity to work on cutting-edge AI technology.
• Collaborative and innovative work environment.
• Flexible working hours.
• Remote work options available.
• Professional development opportunities.