Machine Learning Engineer

About Keyway Labs:

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:

This is a new role. The team has doubled in size over the past year. ML engineering role focused on pipelines, serving and monitoring. You'll work alongside data scientists rather than replace them. This is a hybrid role. Most of the team is in the Houston office Tuesday to Thursday.

Day to day:

• Build training pipelines in PyTorch or TensorFlow and run them on GPUs in AWS, GCP or Azure
• Monitor models for drift, data quality issues and latency regressions
• Productionize data science prototypes without rewriting everything
• Develop RAG pipelines with embeddings, vector stores (pgvector, Pinecone, Weaviate) and rerankers
• Fine-tune open-weight models with LoRA when it beats prompting
• Track experiments, artifacts and versions in MLflow or Weights & Biases
• Document model cards, data sources and known failure modes

You might be a good fit if you:

• 3+ years of software engineering experience, with at least two spent on ML systems in production
• Familiarity with Docker, Kubernetes and a cloud ML platform (SageMaker, Vertex AI, Azure ML)
• Understanding of core ML concepts: overfitting, evaluation metrics, bias in training data

Preferred qualifications:

• CUDA or Triton kernel experience
• Recommendation or ranking systems at scale
• Guardrails, red teaming or safety evaluation for LLM features

Why work with us:

Salary range: $156,000 - $199,000 per year

• Health insurance
• 401(k) with 4% match
• 11 paid holidays
• Medical, dental and vision insurance

Applicants must be authorized to work in the U.S.

Start: Immediately.

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