← All roles

Machine Learning Engineer

Onyx· Engineering
Apply on company site
San Franciscofull-timePosted 10mo ago

About the role

Help improve Onyx's agent and knowledge retrieval capabilities using advanced LLM techniques and NLP, from inception to production, in a collaborative environment based in San Francisco.

Responsibilities

  • Evaluating and implementing LLM based knowledge graphs, advanced RAG approaches (StructRAG, etc.), LLM agents, advances in NLP, multi-modal transformers, advanced information retrieval algorithms
  • Working on users’ experience with the platform through features like learn from feedback, search personalization, SME suggestion, etc.
  • Build a semantic and programmatically useful understanding of the organization's priorities, projects, and people as additional signals to the answering capabilities of Onyx
  • Own the approach from inception to validation to production code
  • Collaborate with Founders and the Head of AI to shape and influence the product and contribute to the AI/ML engineering strategy

Qualifications

  • 3+ years of AI/ML engineering experience building real-world applications
  • In-depth experience with PyTorch/Tensorflow, NLP models, and standard ML algorithms
  • Up to date with new advances such as open source/proprietary LLMs, RAG and agent-frameworks
  • Strong software engineering background capable of building backend features with web frameworks, ORMs and relational DBs
  • Excellent communication skills and ability to collaborate with full stack roles

Skills

Required:LLM-based Knowledge GraphsRAG Approaches (StructRAG)NLP ModelsPyTorch/TensorflowSoftware EngineeringCommunication Skills
Last updated Apr 21, 2026. Interested? Apply on the company site.
Onyx logo

About Onyx

Open Source AI Assistant and Enterprise Search

Onyx is the open source generative AI platform connected to your company’s docs, apps, and people. We ingest and sync from all sources of information (Google Drive, Slack, GitHub, Confluence, Salesforce, etc.) to provide a centralized place for users to ask about anything. Imagine your most knowledgeable co-workers, all-rolled into one, and available 24/7! Check us out on Github: https://www.github.com/onyx-dot-app/onyx Or visit our website: https://www.onyx.app/

open-sourcenlpenterprisesearchai
Company profileWebsite

More open roles at Onyx