Jobs Companies Onyx Machine Learning Engineer

About this Machine Learning Engineer role at Onyx

Onyx · Onsite · San Francisco

About the role

💥 Your impact

Onyx is a popular open source project with hundreds of thousands of users. The project has over 10K stars and over 3K community members across Slack and Discord (these stats may already be out of date when you read this). You’ll have the opportunity to build in the open and your work may be used by millions of people in the future.

💡 About the role

Onyx is the knowledge layer on top of LLMs. Help us improve our agent and knowledge retrieval capabilities to push the frontier on unsolved problems like multi-hop QA, needle in haystack, aggregation type RAG, etc. This is an in-person role based in San Francisco, CA.

You’ll be:

  • 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 direction of the product and contribute to the AI/ML engineering strategy

🚀 You’ll be successful if you…

  • Have 3+ years of AI/ML engineering experience building real-world applications

  • have in-depth experience with PyTorch/Tensorflow, NLP models, and standard ML algorithms

  • Are up date with new advances such as open source/proprietary LLMs, RAG and agent-frameworks

  • Strong software engineering background and capable of building backend features with web frameworks, ORMs and relational DBs

  • Excellent communication skills and ability to collaborate with full stack roles

⭐ Bonus points

  • Familiar with the full stack Typescript/React/NextJS, Python, Postgres

  • Interested in writing technical blogs to establish Onyx is leader in the space

About the interview

  • Non-technical Phone Screen (30 mins)

  • ML Interview (45 mins)

  • Practical Coding Interview (30 mins)

  • Work Trial (3 days in person, fully covered + compensated)

About Onyx

Onyx is the open source GenAI 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!

We believe that every modern team will be adopting knowledge enhanced GenAI within the next 5 years and it is our goal to bring this technology to all the teams of the world.

We raised a $10M seed coming out of YCombinator, backed by Khosla Ventures (early/seed backers of OpenAI, Doordash, GitLab, etc.) and First Round Capital (Notion, Square, Roblox, etc.). Our customers include of the best teams in the world like Netflix, Ramp, Applied Intuition and dozens of others. We also have incredible open source users like Roku, Zendesk, L3Harris and more.

Ready to apply to Onyx?
Apply to Onyx

How this ML Engineer salary compares

This role pays $200,000/yrin line with the typical range for ML Engineer roles.

$169,350 median $239,000 $335,000

Typical range $195,500–$280,000/yr, from 176 comparable ML Engineer listings on JobsRadar (pay annualized to USD). See ML Engineer salary insights →

Similar jobs

Sign up for suggestions tailored to the jobs you open and the searches you save.

More jobs at Onyx

See all jobs at Onyx →

Apply now
🤖

Whoa — hold up

JobsRadar was built for real people having a rough time in their job search — not for automated requests. You're clicking way too fast and you're now temporarily blocked.

Come back later. If you're genuinely job hunting, we've got your back — just act like a human.

Catch your next role the second it’s posted.

Create a free account and we’ll watch the boards for you — the instant a job matches your search, it lands in your inbox or Telegram. No digging, no refreshing.

Create free account

Free forever · takes 30 seconds · already have one?

Get an edge on your job hunt.

Join our Telegram channel for the stuff that helps you land the role — salary benchmarks, the weekly market pulse, and new-feature drops. No spam, just signal.

Join the channel — it's free