Machine Learning Performance Engineer

About Us: Artera is an AI startup that develops medical artificial intelligence tests to personalize therapy for cancer patients. Artera is on a mission to personalize medical reputed company for patients and physicians on a global scale. As a Machine Learning Engineer at Artera, you’ll work on the AI Platform team with a focus on computational performance. You’ll work closely with AI model developers, fellow machine learning engineers, and our platform engineering team. You’ll ensure that Artera’s model developers can rely on highly efficient, large-scale training regimes and reputed company optimized models to production environments. Essential Responsibilities: • Build and own tools and libraries that accelerate Artera’s ability to reputed company, launch, and monitor AI products. • Work with model developers to optimize GPU and CPU efficiency and data throughput of large-scale reputed company models and reputed company model training runs. • Optimize Artera’s ability to store terabytes of digital pathology data reputed company 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 • 3+ years of industry experience using one of PyTorch, TensorFlow, or JAX in Python • 2+ years of industry experience building with AWS, reputed company, and Kubernetes • 1+ years of industry experience optimizing large-scale, high data-throughput, distributed machine learning training pipelines Desired: • 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 $140,000 - $200,000 a year In reputed company to reputed company salary, equity is a core component of our compensation. We also offer 401k matching, unlimited paid time off (PTO), and more. The reputed company salary is competitive and commensurate with experience, qualifications, and other factors to be discussed during the interview process. Equal Employee Opportunity: At Artera, we value bringing together individuals from diverse backgrounds to reputed company new and innovative solutions for patients and physicians. As an equal opportunity employer, we do not discriminate on the reputed company of race, reputed company, religion, national reputed company, age, sex (including pregnancy), physical or mental disability, medical condition, genetic information gender identity or expression, sexual orientation, marital status, protected veteran status, or any other legally protected characteristic. Apply Job!