Onyx

Machine Learning Engineer

Onyx
Apply
over 1 year ago
San Francisco, CA, USAMid Level
H1B Sponsor

Base Salary

$150k - $250k/yr

Responsibilities

  • Evaluate and implement LLM-based knowledge graphs, advanced RAG approaches such as StructRAG, LLM agents, advances in NLP, multimodal transformers, and advanced information retrieval algorithms.
  • Develop user-facing platform features such as learning from feedback, search personalization, and subject-matter-expert suggestions.
  • Build a semantic and programmatically useful understanding of organizational priorities, projects, and people to improve answering capabilities.
  • Own approaches from inception through validation and production code.
  • Collaborate with the founders and Head of AI to shape product direction and AI/ML engineering strategy.

Requirements

  • At least three years of AI/ML engineering experience building real-world applications.
  • In-depth experience with PyTorch or TensorFlow, NLP models, and standard machine learning algorithms.
  • Current knowledge of open-source and proprietary LLMs, RAG, and agent frameworks.
  • Strong software engineering background with the ability to build backend features using web frameworks, ORMs, and relational databases.
  • Excellent communication and collaboration skills.
  • Familiarity with the full-stack TypeScript ecosystem, React, Next.js, Python, and Postgres is a bonus.
  • Interest in writing technical blogs is a bonus.

Benefits

  • In-person role based in San Francisco, California.
  • Three-day in-person work trial is fully covered and compensated.