Skip to content
OnyxOnyxSan Francisco, CA

Machine Learning Engineer

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.

150k – 250k/yr
On-site3+ YOEML Engineering

About the role

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

Skills

PyTorchTensorFlowNLPLLMsRAGLlm AgentsKnowledge GraphsStructragMulti-Modal TransformersInformation RetrievalPythonPostgresReactNext.jsTypeScript

Similar roles

ML Engineering jobs
Applied Intuition

Perception & Fusion Engineer

Applied IntuitionArlington, VA

Develop and integrate real-time sensor perception and fusion algorithms for autonomous vehicles across land, air, sea, and space domains. Requires MS/PhD or 5+ years experience with multi-modal sensors (EO/IR/radar), ML deployment, and DoD clearance eligibility.

150k – 220k/yr
On-site5+ YOEML Engineering
Deepgram

Applied ML Engineer

DeepgramCalifornia

Own the research-to-production pipeline at Deepgram, turning experimental speech ML models into reliable, scalable production services. Partner with researchers on robust workflows, automated release gates, inference optimization, and feedback loops across hybrid GPU infrastructure.

150k – 220k/yr
Remote5+ YOEML Engineering
Deepgram

Backend Engineer

DeepgramCalifornia

Backend Engineer building and optimizing Deepgram's core inference services for speech processing, including networking, audio transcoding, latency/memory optimization, and distributed compute orchestration. Requires 3+ years experience with Rust (or C/C++) and Python.

150k – 220k/yr
Remote3+ YOEML Engineering
Distyl AI

Research Engineer, Post-Training

Distyl AISan Francisco, CA +1

Research Engineers at Distyl build and productionize post-training techniques (fine-tuning, RLHF, reward models, evals) to improve reliability and behavior of compound AI systems for enterprise customers. Requires strong applied ML experimentation skills and ownership of real-world outcomes.

150k – 250k/yr
HybridML Engineering
Distyl AI

Research Engineer, Agents

Distyl AISan Francisco, CA +1

Research Engineers at Distyl build and productionize reliable agentic AI systems and compound architectures for enterprise workflows. They design agents, develop evaluation frameworks, run experiments on reasoning and failure modes, and integrate into customer environments.

150k – 250k/yr
HybridML Engineering