ML Engineer

About us

Most of what we build is back-office software for property managers, including lease tracking, work orders and tenant payments. The product is a mobile app plus a React web dashboard, with a Python and PostgreSQL backend on AWS.

The role

Wagongrid has an opening for an ML Engineer. We're looking for an engineer to build retrieval-augmented generation features on hosted and open-weight language models, with proper evaluation. You can work from home anywhere in the US. We have no office requirement. Standard office hours, Monday to Friday, with flexibility around them. Most communication happens in Slack, with short video calls when needed.

Day to day

• Write clean, tested Python and review teammates' code.
• Cut inference cost with quantization, batching and caching.
• Monitor models for drift, data quality issues and latency regressions.
• Set up evaluation harnesses and regression tests for model quality.
• Orchestrate pipelines with Kubeflow, SageMaker Pipelines, Vertex AI or Airflow.
• Deploy models behind low-latency APIs with FastAPI, Triton, TorchServe or vLLM.

About you

• 3-5 years of software engineering experience, with at least two spent on ML systems in production.
• Strong Python and solid engineering habits (testing, packaging, code review)
• Experience deploying models as services and monitoring them.
• Understanding of core ML concepts: overfitting, evaluation metrics, bias in training data.
• Familiarity with Docker, Kubernetes and a cloud ML platform (SageMaker, Vertex AI, Azure ML)
• Hands-on experience with PyTorch, TensorFlow or JAX.

Also helpful

• CUDA or Triton kernel experience.
• Computer vision background (OpenCV, detection, segmentation)

Perks and benefits

• Health insurance.
• 401(k) with company match.
• Paid time off.
• 11 paid holidays.
• Paid sick leave.

Most hires start within two to three weeks of an offer.

Start: Flexible.

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