Flock Safety is seeking a Staff Machine Learning Engineer to lead the advancement of their core embedding-based retrieval systems and improve performance, generalization, and cross-modal alignment.
Requirements
- 7+ years of industry experience in Machine Learning with a focus on representation learning, multimodal modeling, or embedding-based retrieval
- Deep domain knowledge in at least one area: computer vision, natural language processing, or recommendation systems
- Strong proficiency in PyTorch, with experience fine-tuning foundation models and adapting pretrained vision-language models to real-world tasks
- Experience with embedding-based retrieval, including contrastive learning, multimodal alignment, and designing evaluation methods for vector similarity search and embedding quality
- Solid engineering fundamentals in Python, with familiarity in Git, SQL, and Bash
- Familiarity with model compression techniques, such as distillation, quantization, and architecture pruning
- Experience with vector search infrastructure, including provisioning, maintaining, and querying large-scale vector databases
Responsibilities
- Lead the advancement of core embedding-based retrieval systems
- Fine-tune and extend multimodal models to improve performance, generalization, and cross-modal alignment
- Unify text and image representations, improve model performance, and ensure extensibility across evolving product use cases
- Customize and extend model architectures, training loops, loss functions, and data pipelines to deliver impact
- Work on embedding-based retrieval, including contrastive learning, multimodal alignment, and designing evaluation methods for vector similarity search and embedding quality
- Automate systems for training, evaluation, testing, and model release
- Own long-term maintenance and become a leader for the team offering hands-on help
Other
- Meet the team & cross-functional stakeholders
- Gain familiarity and performing R&D
- Begin to automate the systems for training, evaluation, testing, and model release
- Own long-term maintenance and become a leader for the team offering hands-on help
- Begin exploratory work
- Bachelor's degree or higher in Computer Science or related field
- Ability to work independently and navigate ambiguity
- Track record of solving open-ended modeling problems
