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.
About the job
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
PyTorch, TensorFlow, NLP, LLMs, RAG, Llm Agents, Knowledge Graphs, Structrag, Multi-Modal Transformers, Information Retrieval, Python, Postgres, React, Next.js, TypeScript
Similar jobs
ML Engineering jobs
LangChain
New York, NYBuild and teach reliable AI agent systems through customer workshops, technical content, guidance, and reference implementations. The role requires strong Python and agent-development experience plus a background delivering customer-facing technical training.
Roboflow
San Francisco, CA
Build reinforcement-learning environments, evaluations, datasets, and scalable infrastructure for frontier AI capabilities. The role suits a high-agency generalist engineer with experience in agents, evaluations, or RL workflows and strong communication skills.
Beacon Biosignals
Boston, MADevelop and productionize machine- and deep-learning algorithms for biosignal and EEG data used in medical devices, clinical development, and diagnostics. The role requires 4+ years of industry experience, DSP and statistics expertise, PyTorch proficiency, and familiarity with regulated environments and production ML practices.
Applied Intuition
Sunnyvale, CA
Develop and deploy ML-first behavior prediction and planning systems for autonomous vehicles, forecasting the motion and interactions of road users. Requires a bachelor's degree, deep learning lifecycle expertise, and at least three years of production software experience with C++ or Python.
Fab2
Austin, TXBuild the AI platform behind fab2, including model infrastructure, agent systems, evaluations, and tools for engineering and fab operations. The role requires strong production software engineering skills and comfort working across frontend, backend, infrastructure, and data.