Remote Machine Learning Engineer (Platform)

Description:

• Artera is an AI startup focused on developing medical artificial intelligence tests to personalize therapy for cancer patients.
• The Machine Learning Engineer will work on the AI Platform team, concentrating on creating scalable and efficient pipelines for data processing and model training.
• The role involves collaboration with AI model developers, fellow machine learning engineers, and the platform engineering team.
• Responsibilities include building and owning tools and libraries to enhance Artera’s ability to develop, launch, and monitor AI products.
• The engineer will work to optimize GPU and CPU efficiency and data throughput for large-scale foundation models and downstream model training runs.
• The position requires optimizing the storage and processing of terabytes of digital pathology data for large-scale training regimes.
• The engineer will ensure that Artera’s observability infrastructure provides insights for ongoing performance optimization across the model landscape.

Requirements:

• Candidates must have 4+ years of industry software engineering experience.
• A minimum of 3+ years of experience using PyTorch, TensorFlow, or JAX in Python is required.
• Applicants should have at least 2+ years of experience building with AWS, Docker, and Kubernetes.
• A minimum of 1+ years of experience optimizing large-scale, high data-throughput, distributed machine learning training pipelines is necessary.
• Desired qualifications include experience with Terraform and SqlAlchemy.
• Familiarity with ML orchestration frameworks such as Ray, Kubeflow, Metaflow, MLFlow, Flyte, Dagster, Argo Workflow, or Prefect is preferred.
• Experience in deploying and maintaining infrastructure for machine learning training and production inference is also desired.
• Knowledge of inference optimization techniques such as TorchScript, ONNXRuntime, DeepSpeed, or AWS Neuron is a plus.

Benefits:

• The salary for this position ranges from $140,000 to $180,000 a year, with compensation being competitive and based on experience and qualifications.
• Equity is a core component of the compensation package.
• Artera offers 401k matching to support employees' retirement savings.
• Employees enjoy unlimited paid time off (PTO) to promote work-life balance.
• The company is committed to equal employment opportunity and values diversity in its workforce.