About the Role
This role focuses on developing state-of-the-art information retrieval models as part of the Search team, with responsibilities spanning training embedding and reranker models, advancing semantic search techniques, and integrating innovations into production systems to transform how people interact with search.
Responsibilities
- Design, train, and refine next-generation search models to push the boundaries of retrieval performance.
- Collect high-quality datasets for retrieval tasks and streamline data pipelines for efficient model training and evaluation.
- Partner closely with model serving engineers to ensure fast, reliable inference performance.
- Work with product teams to co-develop practical, scalable solutions for real-world applications.
- Participate in research partnerships with academic and industry collaborators and publish findings in leading conferences and journals.
- Join at a critical growth stage, help define technical direction, take ownership of projects, and adapt to evolving challenges.
Requirements
- Strong proficiency in Python and experience with machine learning frameworks such as PyTorch, TensorFlow, TF-Serving, JAX, or XLA/MLIR.
- Knowledge of information retrieval techniques and hands-on experience training and applying retrieval models.
- Experience integrating Large Language Models into training or evaluation workflows.
- Excellent communication abilities and proven problem-solving skills in technical environments.
Nice to Have
- Background in creating training or evaluation datasets for real-world retrieval use cases.
- Programming skills in languages like C++ or Golang beyond Python.
- Experience implementing large-scale distributed training systems using GPU clusters.
Benefits
- Weekly lunch stipend of $75 or £75, or equivalent in local currency.
- Comprehensive health and dental coverage, including dedicated mental health support funding.
- Retirement savings plans with matching contributions: RRSP, 401K, or Pension Scheme.
- 100% paid parental leave top-up for up to six months, available to either parent.
- Annual enrichment allowance covering arts and culture, fitness, wellness, personal time, and workspace upgrades.
- Dedicated education budget for courses, conferences, and professional coaching.
- Six weeks of paid vacation per year (30 working days).
- Funding to visit other offices if remote, plus attendance at annual company offsites.
- Daily lunch, snacks, and social events for office-based employees.
- Co-working space benefit for remote employees located far from physical offices.
- One-time $500 stipend for home office setup.
Compensation
Not specified
Work Arrangement
Remote (Worldwide) — Toronto, San Francisco, New York City, London, Paris, Montreal, Seoul, Germany
Who are we?
The company is a leading security-first enterprise AI organization focused on building advanced foundation models and end-to-end products that solve real business challenges. It trains and deploys frontier AI models for enterprise clients and emphasizes responsible, impactful AI adoption. The team consists of passionate researchers, engineers, and designers committed to advancing model capabilities and customer value. With co-headquarters in Toronto and San Francisco, it operates globally across key hubs including London, New York City, Montreal, Seoul, Germany, and Paris.
Why this role?
The team seeks skilled professionals to advance information retrieval by developing cutting-edge embedding and reranker models. You will play a key role in redefining search experiences through semantic search innovation, working with novel technologies and integrating solutions into scalable infrastructure. This position offers the chance to experiment, build, and deploy new ideas while collaborating across disciplines. Ideal candidates are deeply interested in search technology and have a solid background in retrieval systems and cross-functional teamwork.
How and Where We Work
The company supports remote work globally, with physical offices in Toronto, San Francisco, New York City, London, Paris, Montreal, and additional locations planned. Office-based employees enjoy daily meals, snacks, and regular team events. Remote employees receive access to co-working spaces in their cities. All employees are provided a $500 home office stipend to support effective remote setups.
Other
- AI-assisted tools may be used to evaluate applicants against role criteria.
- Candidates needing accommodations during the hiring process can submit an Accommodations Request Form to ensure accessibility and support.
Not specified
