# Machine Learning Engineer

**Company:** [Onyx](https://hotfix.jobs/companies/onyx)
**Location:** San Francisco, CA
**Role:** ML Engineering
**Salary:** $150k – $250k/yr
**Experience:** 3+ years
**Skills:** PyTorch, TensorFlow, NLP, LLMs, RAG, Llm Agents, Knowledge Graphs, Structrag, Multi-Modal Transformers, Information Retrieval, Python, Postgres, React, Next.js, TypeScript
**Posted:** 2025-04-21

> Develops advanced LLM-based knowledge graphs, RAG techniques, agents, and NLP features to enhance Onyx's AI knowledge retrieval platform. Requires 3+ years AI/ML experience with PyTorch/TensorFlow and strong software engineering skills.

## Job Description

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

## Requirements
- 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 and capable of building backend features with web frameworks, ORMs and relational DBs
- Excellent communication skills and ability to collaborate with full stack roles

## Nice-to-haves
- Familiar with the full stack Typescript/React/NextJS, Python, Postgres
- Interested in writing technical blogs to establish Onyx is leader in the space

## Similar roles

- [Perception & Fusion Engineer](https://hotfix.jobs/jobs/8655c0c2-c78a-420d-b397-b01bb4bef878) - Applied Intuition - Arlington, VA - $150k – $220k/yr
- [Applied ML Engineer](https://hotfix.jobs/jobs/3016b55f-07d3-486b-ba7c-0c68f9819cd0) - Deepgram - Remote - $150k – $220k/yr
- [Backend Engineer](https://hotfix.jobs/jobs/ab7e0e7b-91ae-4435-ae9f-df959e7843f1) - Deepgram - Remote - $150k – $220k/yr
- [Research Engineer, Post-Training](https://hotfix.jobs/jobs/09d480a0-f2fe-4418-8549-14c5c9fa2aa0) - Distyl AI - San Francisco, CA - $150k – $250k/yr
- [Research Engineer, Agents](https://hotfix.jobs/jobs/bb345787-77ae-40a4-8b65-895e6a979c5e) - Distyl AI - San Francisco, CA - $150k – $250k/yr

**Apply:** https://hotfix.jobs/jobs/7c512a2f-9161-4506-8740-0ebed20915ef
**Canonical:** https://hotfix.jobs/jobs/7c512a2f-9161-4506-8740-0ebed20915ef