Machine Learning Engineer, Digital Experience
Pure Storage13 hours ago
Santa Clara, CA, USAMid Level / Senior
H1B sponsor
Base Salary
$180k - $270k/yr
Responsibilities
- Take machine learning models from prototype to production by building reliable, scalable training, serving, and inference pipelines.
- Design and maintain data pipelines, feature stores, and workflow orchestration for clean, timely, and well-tested model inputs.
- Build monitoring for model performance, data drift, and pipeline health, and respond to production failures.
- Build and validate machine learning models and statistical approaches when needed in collaboration with data scientists.
- Work with Data Scientists, Data Engineers, and Software Engineers to turn business questions into production systems.
Requirements
- Bachelor’s, master’s, or Ph.D. in Computer Science, Data Science, Engineering, Statistics, or a related field, or equivalent practical experience.
- 3–5 years of industry experience in data engineering, ML engineering, or a hybrid data science/engineering role, including a track record of shipping models to production.
- Strong software engineering fundamentals in Python and SQL, including production-quality, well-tested code.
- Hands-on experience with a workflow orchestration tool such as Airflow or Dagster.
- Experience working in a cloud-native environment using AWS, GCP, or Azure.
- Working knowledge of machine learning and statistical modeling, with familiarity with Scikit-Learn or PyTorch and the ability to build or extend models.
- Clear communication skills for explaining technical work to non-technical stakeholders and the ability to work collaboratively through ambiguity and changing priorities.
Benefits
- Primarily in-office work from the Santa Clara office, except during PTO, work travel, or other approved leave.
- Flexible time off, wellness resources, and company-sponsored team events.
- The role may be eligible for incentive pay and/or equity.
- The company provides accommodations for candidates with disabilities during the hiring process.
Categories
Data EngineeringML Engineering
About Pure Storage
Pure Storage builds all-flash data storage systems and software for enterprises, including FlashArray and FlashBlade, and offers storage delivered as-a-service for on‑prem and cloud workloads. Its subscription model (Evergreen/Pure as‑a‑Service) supports databases, virtualization, analytics, and AI. Founded in 2009 and headquartered in Santa Clara, California, Pure is a public company listed on the NYSE (PSTG).