Data Engineer – AI/ML Data Infrastructure
Yogi Careers · Boston, MA · 2 mo ago
Information TechnologyFull-time
Responsibilities
- Develop and maintain feature stores and ML-ready datasets
- Automate data preprocessing pipelines for ML model training and evaluation
- Collaborate with ML engineers to enable scalable experimentation workflows
- Monitor and improve data reliability, lineage, and reproducibility
Requirements
- BS/MS in Computer Science, Data Engineering, or similar
- Experience with ML platforms (Databricks, AWS Sagemaker, Vertex AI)
- Strong Python and SQL skills, with familiarity in Spark or Dask
- Experience with Airflow, MLflow, or Kubeflow pipelines
- Solid understanding of MLOps, data validation, and model versioning