Member of Technical Staff (ML)

Engineering, Mountain View Office

Samaya AI is building knowledge discovery tools for domain experts using AI and Large Language Models. Our platform enables users to answer complex questions and uncover hidden insights from large sets of knowledge-intensive documents.

Founded by Maithra Raghu (formerly at Google Brain) and Fabio Petroni (formerly at Meta AI), Samaya AI has incredible firepower in terms of researchers, engineers, and developers with a breadth of AI experience, including Stanford, Meta AI, Google, Microsoft, and more. 

Members of our technical staff develop and deploy large-scale language models (LLMs) which power our knowledge discovery platform. In close collaboration with our other team members, you will build, maintain, and evaluate the core ML components of our knowledge discovery pipelines. Potential skills include: data curation, model training, evaluation, inference optimization, scalable deployment, and research in new ML methods.  

Your Opportunity

  • Research and propose cutting-edge neural solutions that solve real-world problems
  • Train, deploy, and evaluate LLMs at scale
  • Design and implement evaluation pipelines to ensure accuracy and efficiency
  • Optimize inference and training infrastructure for latency and costs

Your first 3 months 

  • You have a strong grasp of Samaya’s ML goals with relation to short-term product focus and longer-term research bets.
  • You have helped define evaluation criteria for critical ML goals.
  • Your contributions have directly improved the quality/latency of various ML tasks in our system.
  • You are finding opportunities to work on interesting and impactful ML problems, with highly skilled and knowledgeable colleagues.

What you’ll bring

  • Demonstrated ML expertise, in researching, training, deploying, and optimizing LLMs and other models/techniques for knowledge-intensive tasks.
  • Strong programming skills, with proficiency in Python and experience with ML libraries/techniques.
  • Proven track record of working with LLMs and neural information retrieval.
  • Independence, and a willingness to take ownership of tasks
  • Passion for solving challenging problems
  • PhD strongly preferred

What we provide

  • Flexible and hybrid work environment (typically spend a minimum of 2 days in the office per week)
  • Comprehensive health insurance coverage, including medical, dental, vision, and short-term disability
  • Opportunity to rest and recharge with unlimited PTO
  • Support your long-term financial well-being with 401K (US) and Pension (UK)
  • Create your work setup with your office equipment allowance
  • Travel to other locations is encouraged to foster team cohesion and maintain Samaya’s unique culture

If you're motivated to make a difference and possess the drive to succeed but are not sure you meet all the criteria, why not give us a try? 
We are committed to ensuring an equitable selection process for everyone and welcome applicants from varied backgrounds to enrich our team. If you require accommodations or adjustments during our recruitment process, please inform us.

*Samaya AI is an Equal Opportunity Employer. We do not discriminate on the basis of race, color, religion, sex (including pregnancy and gender identity), national origin, political affiliation, sexual orientation, marital status, disability, genetic information, age, membership in an employee organization, retaliation, parental status, military service, or other non-merit factor. 

We pride ourselves on working collaboratively in an inclusive, truth-seeking environment, dedicated to changing the way professionals access and use knowledge through generative AI. Join us in shaping the future of knowledge and AI technology! 

Apply now