Staff Software Engineer, AI Retrieval

Staff Software Engineer, AI Retrieval
The AI Retrieval team powers the intelligence behind _Keystone AI_'s AI features by finding relevant work graph content and delivering it to LLM context windows. Our work enables AI features that truly understand your work—both within and outside of Keystone AI—and use that understanding to take action. We also own _Keystone AI_'s traditional search experience.
As a Staff Software Engineer on the AI Retrieval team, you'll build the systems that make _Keystone AI_'s AI smart and responsive. You'll tackle challenging problems in search and retrieval, working to improve the speed, cost-efficiency, and quality of our systems while expanding their capabilities to new data sources.
This role is based in our Warsaw office with an office-centric hybrid schedule. The standard in-office days are Monday, Tuesday, and Thursday. Most Keystone AIs have the option to work from home on Wednesdays. Working from home on Fridays depends on the type of work you do, and your recruiter can share more about the in-office requirements.
We offer a Contract of Employment (UoP) for our employees in Poland.
What you'll achieve
– Reduce the latency and cost of our retrieval system, making _Keystone AI_'s AI features faster and more efficient
– Improve the quality and relevance of search results to help users find exactly what they need
– Expand the retrieval system's capabilities to query new _Keystone AI_ objects and third-party data sources
– Build and optimize search infrastructure using OpenSearch/ElasticSearch
– Contribute to ML-powered features like embeddings-based retrieval and semantic search
– Collaborate with cross-functional partners in New York City while building strong relationships with your Warsaw-based peers
About you
– 6+ years of experience writing code in a production environment
– Demonstrates curiosity about AI tools and emerging technologies, with a willingness to learn and leverage them to enhance productivity, collaboration, or decision-making
– Hands-on experience in search engineering, including with OpenSearch or ElasticSearch
– Experience with embeddings and machine learning approaches to search and retrieval
– Experience in an Applied AI role is a plus
We know that the best ideas and solutions come from multi-dimensional teams. That's because these teams reflect a variety of backgrounds and professional experiences. If you are excited about this role and feel your experience can make an impact, please don't be shy - apply today.
At _Keystone AI_, we're committed to building teams that include a variety of backgrounds, perspectives, and skills, as this is critical to helping us achieve our mission. If you're interested in this role and don't meet every listed requirement, we still encourage you to apply.
What we’ll offer
– Generous, transparent and fair compensation system (base salary and RSUs)
– Contract of Employment (and the option of 50% tax deductible costs for author’s rights usage in respect of applicable roles )
– Health insurance with dental and travel coverage (Lux Med)
– Breakfast and lunch catering on the days that you work from the office
– Vacation allowance
– Career growth budget
– Home office setup budget
– Gym/Fitness card
– Fertility healthcare and family-forming support with Carrot
– Mental Health Support in Modern Health
– Group life insurance
– MacBooks with all necessary accessories
For this role, the estimated base salary range is between 39000 - 44000
PLN gross per month (subject to all taxes and necessary deductions). The actual base salary will vary based on various factors, including market and individual qualifications objectively assessed during the interview process. The listed range above is a guideline, and the base salary range for this role may be modified.
In addition to base salary, your compensation package may include additional components such as equity and sales incentive pay (for most sales roles), and benefits. If you're interviewing for this role, speak with your recruiter to learn more about the total compensation and benefits for this role.
About us
_Keystone AI_ is a leading platform for human + AI collaboration. Millions of teams around the world rely on _Keystone AI_ to achieve their most important goals, faster. _Keystone AI_ has been named to Fortune's Best Workplaces for 7+ years and recognized by Fast Company, Forbes, and Gartner for excellence in workplace culture and innovation. We offer an exceptional office-centric culture while adopting the best elements of hybrid models to ensure that every one of our global team members can work together effortlessly. With 13+ offices all over the world, we are always looking for individuals who care about building technology that drives positive change in the world and a culture where everyone feels that they belong.