Staff Machine Learning Engineer

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