about 4 hours ago
Menlo Park, CA, USA
Senior / Mid Level
H1B Sponsor
Base Salary
$196k - $230k/yr
Responsibilities
- Design, build, and maintain scalable systems for deploying, monitoring, and managing Machine Learning models in production.
- Partner with ML practitioners to streamline workflows, integrate internal ML libraries, and optimize model performance.
- Contribute to the development and scaling of the feature store for efficient feature retrieval.
- Implement robust observability for model performance, data pipelines, and feature freshness.
- Manage and optimize cloud compute resources to support cost-effective training and inference across AWS.
Requirements
- 4+ years of software engineering experience, ideally within ML infrastructure, data engineering, or model operations.
- Hands-on experience with model serving, distributed systems, or production ML workflows.
- Familiarity with modern ML infrastructure tools like Ray, Kubeflow, SageMaker, TensorFlow Serving, and Triton.
- Strong proficiency in Python, C++, or similar languages, and experience with ML frameworks like TensorFlow or PyTorch.
- Hands-on experience with large-scale search systems and distributed retrieval engines.
- A collaborative mindset with an interest in building reliable platforms that support company-wide impact.
Benefits
- Challenging, high-impact work to grow your career.
- Performance driven compensation with multipliers for outsized impact, bonus programs, equity ownership, and 401(k) matching.
- 100% paid health insurance for employees with 90% coverage for dependents.
- A highly flexible benefits spending account for wellness, learning, and more.
- Employer-paid life & disability insurance, fertility benefits, and mental health benefits.
- Time off to recharge including company holidays, paid time off, sick time, parental leave, and more.
- Exceptional office experience with catered meals, events, and comfortable workspaces.
Tech Stack
AWSC++ElasticsearchPythonPyTorchTensorFlow
Categories
AI & MLData Engineering