Onebrief Logo

Onebrief

Senior AI/ML Engineer, Knowledge & Retrieval Systems

Reposted 19 Days Ago
Remote
Hiring Remotely in USA
Mid level
Remote
Hiring Remotely in USA
Mid level
Design and build hybrid retrieval systems, develop pipelines for operational knowledge, and architect knowledge graphs while collaborating with experts.
The summary above was generated by AI
About Onebrief

Onebrief is collaboration and AI-powered workflow software designed specifically for military staffs. By transforming this work, Onebrief makes the staff as a whole superhuman - meaning faster, smarter, and more efficient.

We take ownership, seek excellence, and play to win with the seriousness and camaraderie of an Olympic team. Onebrief operates as an all-remote company, though many of our employees work alongside our customers at military commands around the world.

Founded in 2019 by a group of experienced planners, today, Onebrief’s team spans veterans from all forces and global organizations, and technologists from leading-edge software companies. We’ve raised $123m+ from top-tier investors, including Battery Ventures, General Catalyst, Insight Partners, and Human Capital, and today, Onebrief is valued at $1.1B. With this continued growth, Onebrief is able to make an impact where it matters most.

Role Overview

We're seeking a Machine Learning Engineer with a deep understanding of information retrieval, knowledge representation, and edge-deployable ML systems.

In this role you will work toward transforming complex, interconnected military operational plans into actionable, queryable knowledge. You'll lead the design and implementation of scalable systems for chunking, indexing, and retrieving rich data from multiple modalities. Your solutions will enable fast, reliable information retrieval to augmented Generative AI systems.

Expect to architect hybrid retrieval pipelines that blend semantic search, keyword-based methods, and graph reasoning, optimize embeddings for specialized content, and build resilient systems that power rapid decision-making.

We're looking for someone with hands-on experience building real-world retrieval and knowledge-driven systems.

What You'll Do

  • Design and build hybrid retrieval systems that combine semantic, symbolic, and graph-based methods

  • Develop pipelines to encode and retrieve operational knowledge using LLMs, vector databases, and custom chunking/indexing strategies

  • Build and optimize retrieval-augmented generation (RAG) systems for high-stakes environments

  • Architect knowledge graphs and integrate them into retrieval workflows

  • Collaborate with ML, product, and domain experts to transform requirements into deployable solutions

Key Technologies

  • Vector Databases, Hybrid Search Pipelines

  • Embeddings & Transformer-based models

  • Knowledge Graphs (Neo4j, RDF, SPARQL, custom schemas)

  • Python, Distributed Systems, ETL pipelines

  • Docker, Kubernetes, Edge Computing platforms

Qualifications

Required:

  • B.S. in Computer Science, Engineering, or equivalent practical experience

  • 2–4 years of experience in applied ML, information retrieval, or knowledge systems

  • Strong Python programming skills

  • Experience with semantic search, vector stores, and retrieval system design

  • Comfort with ETL workflows and structured, domain-specific datasets

  • Understanding of distributed systems and performance trade-offs

  • Familiarity with testing and evaluating information retrieval systems

  • Understanding of security considerations in data handling and system design

Preferred

  • Experience designing chunking/indexing pipelines for large, domain-specific datasets

  • Experience designing or deploying knowledge graphs in real-world systems

  • Experience with offline-capable and edge-deployable ML systems

  • Familiarity with containerization and orchestration tools (Docker, Kubernetes)

  • Exposure to geospatial data and reasoning systems

  • Background in defense, national security, or other mission-critical domains

  • Understanding of LLM prompt engineering, context window optimization, and RAG techniques

  • Advanced degree (M.S. or PhD) in a relevant field is a plus

Working Style:

  • First principles thinking with high ownership mentality

  • Strong communication and collaboration skills

  • Bias for action - you deliver working systems in imperfect conditions

  • Comfortable working autonomously in a fast-moving startup environment

Compensation Range: $200K - $230K


#BI-Remote

Top Skills

Docker
Edge Computing
Embeddings
Hybrid Search Pipelines
Knowledge Graphs
Kubernetes
Neo4J
Python
Rdf
Sparql
Transformer-Based Models
Vector Databases

Similar Jobs at Onebrief

4 Hours Ago
Remote
USA
Senior level
Senior level
Software • Defense
The role requires expertise in cybersecurity focusing on cloud-native technologies, using various security tools, and collaborating in a remote environment to secure IT ecosystems.
Top Skills: Aws SecurityBashIdentity And Access Management (Iam)IntuneJAMFMobile Device Management (Mdm)OktaPowershellPythonSecure Access Service Edge (Sase)Security Information And Event Management (Siem)SplunkWorkspace OneZscaler
4 Hours Ago
Remote
USA
Senior level
Senior level
Software • Defense
The Staff Backend Engineer will design and maintain backend services, optimize data models, and ensure system reliability and performance, collaborating cross-functionally to deliver high-impact solutions.
Top Skills: AWSAzureDockerGCPGoGraphQLGrpcJavaKubernetesNode.jsPythonRest
4 Hours Ago
Remote
USA
Senior level
Senior level
Software • Defense
Design, build, and maintain backend services for a platform. Ensure performance and reliability in production systems, collaborate on architecture, and oversee long-term strategy.
Top Skills: AWSAzureDockerGCPGoGraphQLGrpcJavaKubernetesNode.jsPythonRest

What you need to know about the Ottawa Tech Scene

The capital city of Canada and the nation's fourth-largest urban area, Ottawa has proven a rapidly growing global tech hub. With over 1,800 tech companies, many of which are leaders in their sectors, the city's tech talent now makes up more than 13 percent of its total workforce. This growth is driven not only by the big players like UL Solutions and Dropbox, but also by a thriving startup ecosystem, as new businesses emerge to follow in the footsteps of those that came before them.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account