Associate ML Engineer - MLOps

Our main product is a payroll and timekeeping app for hourly workforces, sold by subscription and supported from Houston. The product is a mobile app plus a React web dashboard, with a Python and PostgreSQL backend on AWS.

About the role:

Keyway Labs has an opening for an Associate ML Engineer - MLOps. ML engineering role focused on pipelines, serving and monitoring. You'll work alongside data scientists rather than replace them. Remote, with occasional travel to Houston (a few times a year).

Main duties:

• Track experiments, artifacts and versions in MLflow or Weights & Biases.
• Orchestrate pipelines with Kubeflow, SageMaker Pipelines, Vertex AI or Airflow.
• Deploy models behind low-latency APIs with FastAPI, Triton, TorchServe or vLLM.
• Write clean, tested Python and review teammates' code.
• Build training pipelines in PyTorch or TensorFlow and run them on GPUs in AWS, GCP or Azure.
• Set up evaluation harnesses and regression tests for model quality.

Skills and experience:

• 0-2 years of software engineering experience, with at least two spent on ML systems in production.
• BS or MS in computer science, engineering or a related field.
• Familiarity with Docker, Kubernetes and a cloud ML platform (SageMaker, Vertex AI, Azure ML)

What you get:

Expected salary range: $108,000 - $137,000 per year, plus bonus eligibility.

• Health, dental and vision coverage starting on day one.
• 401(k) with a 100% match up to 5%.
• Paid holidays.
• Life insurance.
• Home office stipend.
• Equity.

EOE. We participate in E-Verify.

No recruitment agencies, please.

Start: Immediately.

Demo job created on 11jobs for testing and demonstrations. It is not an actual opening at Keyway Labs.