Staff Machine Learning Engineer at Simplisafe · LumiJobs
senior
$184k–$245k
H1B: yes
Simplisafe Staff Machine Learning Engineer at SimpliSafe leading technical direction for ML data and training platform including large language models. Owns architecture and drives multi-quarter initiatives across teams.
Responsibilities
We are looking for a Staff Machine Learning Engineer to serve as the technical leader for our ML data and training platform. In this role, you will set the technical direction for how SimpliSafe prepares data, trains, fine-tunes, and tracks machine learning models, including large language models. You will own the architecture of our data and training infrastructure, drive multi-quarter initiatives across teams, and raise the engineering bar through design reviews, mentorship, and standards. If you want to shape how ML models get built at scale, we'd love to hear from you.
● Define and own the technical vision, architecture, and multi-quarter roadmap for SimpliSafe's ML data and training platform.
● Architect and lead the delivery of scalable data and feature pipelines that produce reliable, versioned training datasets.
● Build and evolve distributed training infrastructure for classical ML, deep learning, and LLM fine-tuning, including efficient GPU scheduling and utilization.
● Own the experimentation stack, including experiment tracking, model registry, and dataset and model lineage, so that every model is reproducible.
● Develop CI/CD and automation for training pipelines, enabling data scientists to go from idea to trained, validated model quickly and safely.
● Partner with data science, data engineering, product, and security leaders to align the platform with business priorities and turn ambiguous problems into clear, executable plans.
● Set org-wide standards and paved paths for data versioning, training reproducibility, and data governance, and drive their adoption across teams.
● Lead build-vs-buy evaluations and make durable technology decisions on ML tooling, cloud services, and frameworks.
● Drive reliability and cost efficiency of training and data workloads in AWS, including capacity planning and compute and GPU cost optimization.
● Mentor and grow senior and mid-level engineers through design reviews, code reviews, and technical coaching; act as a force multiplier for the team.
● Participate in the team's on-call rotation for platform services the team owns, and lead postmortems for critical incidents.
Qualification
● Deep experience with CI/CD pipelines● Strong proficiency in Python● Experience handling PII data● Knowledge of cloud securityWhat Values You’ll Share
Required
● 8+ years of experience in software engineering, data engineering, or a related field, with at least 4 years focused on ML infrastructure or MLOps.
● Proven track record of technical leadership: designing and shipping ML training platforms, data platforms, or large-scale distributed systems end to end, and leading cross-team initiatives without direct authority.
● Expert-level experience with AWS or similar public clouds, including compute, storage, networking, and container orchestration.
● Deep experience with CI/CD pipelines, Docker, Kubernetes, and infrastructure-as-code tools (e.g., Terraform, CloudFormation).
● Strong proficiency in Python, and hands-on experience with machine learning frameworks (e.g., PyTorch, TensorFlow).
● Hands-on experience with distributed training and large-scale data processing (e.g., Ray, Spark, PyTorch Distributed).
● Deep understanding of experiment tracking, model registries, and data and model versioning (e.g., MLflow, Weights & Biases, DVC).
● Excellent written and verbal communication skills, including writing technical design docs and influencing senior technical and non-technical stakeholders.
● Experience with feature stores and data lakehouse technologies (e.g., Feast, SageMaker Feature Store, Delta Lake, Iceberg).
● Experience building pipelines for large unstructured datasets such as video, images, or audio, including labeling and curation workflows.
● Experience with event-driven or streaming data architectures (e.g., Kafka, Kinesis).
● Knowledge of cloud security, IAM, data privacy, and compliance best practices for ML data and workloads.
What Values You’ll Share
Customer Obsessed - Building deep empathy for our customers, putting them at the core of our work, and developing strong, long-term relationships with them.