About the Role

Job Title: Staff ML Engineer
Location: Zürich
Employment Type: Full-time
Location Type: Hybrid
Department: Engineering

Overview

You want to operate at the cutting edge of AI, building the strongest defenses in the AI cybersecurity market. You will be increasing the scope and quality of our defenses and playing a pivotal role in ensuring that GenAI applications can be deployed securely at scale.

Responsibilities

• Lead the creation of cutting-edge models that perform exceptionally well and operate in real time.
• Continuously identify edge cases for our detectors and work to collect and generate intricate threat datasets to improve model performance.
• Ensure that Lakera can continuously ship the best defenses in the market with a high cadence.
• Shape the future by working on ideas that define the secure use of AI in the decades ahead.

Example Projects

• Extend the scope of one of our detectors to support multiple languages without dropping performance on already supported languages.
• Increase the scope where our defenses perform by 100x (e.g., multi-lingual data, code, etc.).
• Extend modeling and data generation to enable passing additional context to our prompt injection detectors, such as the LLM output and behavioral metadata.
• Red team our model to understand the most pressing vulnerabilities of our detectors and prioritize the data collection and generation required to fix them.

Qualifications

• At least 8 years of experience shipping machine learning models to production, ideally in environments where reliability and quality are key (e.g., autonomous driving, industrial applications, healthcare).
• Most experience productionizing ML models has been in unstructured data spaces such as NLP and vision.
• Experience working on end-to-end development of ML models.
• Experience working on the “long tail” of data distribution, bringing models from prototypes to real-world systems.
• Aspiration to play a pivotal role in defining the future of secure AI within a mission-driven company with ambitious goals.