Senior AI/ML Engineer, Knowledge & Retrieval Systems

Job description

About us

Onebrief is a revolutionary platform for military staff workflows and operational planning. The software is designed to enable smarter, real-time decisions. With unparalleled collaboration features, AI-enhanced tools, and customizable workflows, Onebrief makes staffs superhuman. The expanding roster of customers includes COCOMs and Service Components worldwide.

Founded in 2019 by a group of experienced planners, today, Onebriefโ€™s workforce of 120+ spans veterans from all forces and global organizations, and technologists from leading-edge software giants. Onebriefโ€™s growth is exemplary, having raised $53M+ and counting from leading venture investors.

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

Share this job:
Please let Onebrief know you found this job on Remote First Jobs ๐Ÿ™

Similar Remote Jobs

Benefits of using Remote First Jobs

Discover Hidden Jobs

Unique jobs you won't find on other job boards.

Advanced Filters

Filter by category, benefits, seniority, and more.

Priority Job Alerts

Get timely alerts for new job openings every day.

Manage Your Job Hunt

Save jobs you like and keep a simple list of your applications.

Search remote, work from home, 100% online jobs

We help you connect with top remote-first companies.

Search jobs

Hiring remote talent? Post a job

Frequently Asked Questions

What makes Remote First Jobs different from other job boards?

Unlike other job boards that only show jobs from companies that pay to post, we actively scan over 20,000 companies to find remote positions. This means you get access to thousands more jobs, including ones from companies that don't typically post on traditional job boards. Our platform is dedicated to fully remote positions, focusing on companies that have adopted remote work as their standard practice.

How often are new jobs added?

New jobs are constantly being added as our system checks company websites every day. We process thousands of jobs daily to ensure you have access to the most up-to-date remote job listings. Our algorithms scan over 20,000 different sources daily, adding jobs to the board the moment they appear.

Can I trust the job listings on Remote First Jobs?

Yes! We verify all job listings and companies to ensure they're legitimate. Our system automatically filters out spam, junk, and fake jobs to ensure you only see real remote opportunities.

Can I suggest companies to be added to your search?

Yes! We're always looking to expand our listings and appreciate suggestions from our community. If you know of companies offering remote positions that should be included in our search, please let us know. We actively work to increase our coverage of remote job opportunities.

How do I apply for jobs?

When you find a job you're interested in, simply click the 'Apply Now' button on the job listing. This will take you directly to the company's application page. We kindly ask you to mention that you found the position through Remote First Jobs when applying, as it helps us grow and improve our service ๐Ÿ™

Apply