Skip to content

Staff Machine Learning Engineer

Builds, deploys, and maintains ML models to detect phishing, sensitive data, and threats in emails/drives. Requires 8+ years ML experience, strong data pipeline skills, and familiarity with LLMs/generative AI.

225k – 255kSan Francisco, CAML EngineeringRemote8+ YOE

About the role

Responsibilities

  • Design, build, train, and deploy machine learning models to detect sensitive data and malicious threats (phishing emails).
  • Write production-level code to convert ML models into working pipelines and participate in code reviews.
  • Architect scalable, reliable, and maintainable machine learning pipelines, integrating with existing backend systems.
  • Explore recent advancements in generative AI and LLMs for detection capabilities.
  • Work closely with machine learning engineers, product managers, designers, data scientists, and software engineers.
  • Stay ahead with new algorithms, technologies, and frameworks.
  • Contribute to engineering culture through participation and mentorship.

Requirements

Must Haves

  • B.S., M.S. or Ph.D. in Computer Science or related technical field or relevant work experience.
  • 8+ years (or Ph.D. with 6+ years) of experience in machine learning, data science, or related fields, with at least 3 years in a senior or staff engineering role.
  • Deep understanding of supervised/unsupervised learning techniques and LLMs.
  • Strong experience writing efficient and effective data pipelines.
  • Practical knowledge of building efficient end-to-end ML workflows from conception through deployment and maintenance.
  • Experience with machine learning libraries (e.g., scikit-learn, Pandas).

Nice to Haves

  • Experience in API development on top of FastAPI.
  • Experience tracking text embedding modeling.
  • Strong knowledge of cloud platforms (e.g., AWS, GCP) and containerization tools (e.g., Docker, Kubernetes).

Compensation

$225,000-255,000

Skills

Machine LearningLLMsGenerative AIscikit-learnpandasData PipelinesSupervised LearningUnsupervised LearningFastAPIKubernetesDockerAWSGCPText Embeddings

Similar roles

ML Engineering jobs

Staff Software Engineer, Cribl AI

Staff-level AI/ML engineer building and productionizing generative AI features across backend and frontend for Cribl's observability platform. Requires 6+ years experience, AI/ML and MLOps background, and TypeScript/JavaScript proficiency.

225k – 265kUnited StatesML EngineeringRemote6+ YOELLMsReact

Member of Research Staff, Voleon Securities (New York)

Develops and deploys advanced ML models for financial market prediction and portfolio optimization in a market-making strategy. Requires PhD-level expertise in statistical ML, preferably deep RL, deep learning, optimal control, or causal inference, with strong math and Python skills.

225k – 250kNew York, NYML EngineeringHybridPythonDeep Rl

Applied AI - Senior/ Staff Software Engineer, Lead

Leads development of production AI systems using LLMs, RAG, and multi-agent architectures to enhance clinical workflows, patient risk prediction, and care coordination in healthcare. Ships end-to-end features with cross-functional collaboration in a fast-paced startup.

225k – 260kSan Francisco, CA +1ML EngineeringHybridRAGHl7

Senior Staff Data Scientist, Guest & Host Marketplace AI

Leads AI-driven innovation to analyze guest behavior, detect preferences, and build personalized discovery products in Airbnb's marketplace. Requires causal inference expertise, 9+ years experience, advanced degree, and strong Python/SQL skills.

224k – 280kUnited StatesML EngineeringRemote9+ YOERAI

Staff Machine Learning Engineer, Ads Conversion

Lead technical vision and development of large-scale DNN models for ads conversion modeling, improving user action prediction with low latency. Requires 6+ years building production ML systems at scale and 2+ years leading technical projects.

223k – 390kSan Francisco, CA +2ML EngineeringHybrid6+ YOESQLLLMs