AI Engineer
Build full-stack AI prototypes and agentic systems to pressure-test venture ideas. Requires 3+ years building production AI applications with strong frontend/backend fluency and frontier coding agent expertise.
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 – 250kSan Francisco, CAML EngineeringOnsite3+ YOE
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 jobsMachine Learning Engineer
Build and deploy ML models for entity resolution and knowledge graph expansion on large-scale China-related data. Requires 4+ years clustering ML experience and end-to-end production ML with Python/SQL.
Research Engineer, Post-Training
Research Engineers design and run post-training workflows, build evaluation infrastructure, and turn frontier AI techniques into reliable production systems for enterprise customers. Requires experience with fine-tuning, RLHF, reward modeling, and strong experimentation skills.
Research Engineer, Agents
Research Engineers build and productionize agentic AI systems, designing compound architectures, evaluation frameworks, and reliable execution for enterprise workflows. Requires strong Python skills, systems reasoning, and experience building agents with tools, retrieval, planning, and memory.