Machine Learning Engineer

Jaxon.AI

Lincoln (MA)

On-site

USD 140,000 - 190,000

Full time

14 days+

Get more replies from employers

Send a job-specific resume in minutes.

Benefits offered by this job

Unlimited PTO
Comprehensive benefits: Health, dental
Stock Options
Holidays Off

Job summary

Jaxon.AI is seeking a Machine Learning Engineer to advance NLP and ML capabilities for defense-focused projects in a trusted, high-stakes environment. You will work on Domain-Specific AI Language (DSAIL) and contribute to secure, reliable AI deployments for national security applications.

Responsibilities include collaborating with distributed teams, optimizing data pipelines, and building models beyond cloud APIs.

Qualifications

  • Bachelor’s degree in Computer Science or a related field.
  • Minimum 5 years of experience in ML, NLP, or AI engineering.
  • Strong Python proficiency and ability to work in cloud environments.
  • Experience with LLM deployment, governance, and guardrails a plus.

Responsibilities

  • Collaborate with cross-functional teams to integrate ML models into systems.
  • Optimize data processing, feature engineering, and model evaluation.
  • Develop NLP solutions for unstructured data using Python.
  • Prototype and deploy LLM-based capabilities with scalable architectures.

Skills

Python
NLP
Data processing
Cloud
LLM deployment

Education

Bachelor's degree in Computer Science or related field

Tools

Langraph
LlamaIndex
Docker
Kubernetes
Ollama
Transformers

Job description

Position Name:Machine Learning Engineer

Employment Type:Full-time

Notes: Direct Hire. No third-party assistance is approved for this role.

About The Job:

At Jaxon, we’re focused on making AI trustworthy for mission-critical environments. Our core technology is built to enable the safe deployment of AI in high-stakes settings, particularly across the Department of Defense. While our company supports both commercial and government applications, this role is dedicated to the defense side. As a Machine Learning Engineer at Jaxon, you’ll be embedded on a sensitive project designed to help solidify the U.S. government’s ability to use AI reliably. The work centers on advancing machine learning capabilities to meet the rigor and reliability required for national security operations.

Jaxon’s product, Domain-Specific AI Language (DSAIL), forms the basis of most of our deployments. This role will focus on increasing security and reliability for our customers through implementing and building on DSAIL’s capabilities. A particular emphasis is placed on addressing the fundamental limitations of LLMs — specifically around context and logical reasoning.

The basics of this role requires the ability to:

  • Collaborate Across Teams: Work with cross-functional, geographically distributed teams to integrate ML models into our existing systems and workflows, enhancing product capabilities.
  • Optimize Data Processing and Model Performance: Conduct comprehensive data management including preprocessing, feature engineering, and model evaluation to improve accuracy and efficiency.
  • Technical Proficiency with an NLP focus: Demonstrate solid machine learning engineering experience, particularly with NLP applications and unstructured data in a cloud environment. Fluency in Python is essential.
  • Understand Large Language Models: Intimately familiar with the model deployment process, optimization of LLM parameters for specific behaviors, and a general understanding of LLM functionality and use cases. Previous experience with AI/LLM governance and guardrail development is a big plus.

Soft Skills:

  • Ability to context switch as needed.
  • Ability to communicate findings and recommendations clearly and constructively to peers, management, partners, and customers.
  • Ability to effectively explain highly technical details to less technically focused decision makers.
  • Ability to compartmentalize different projects with different goals.
  • Able to communicate clearly and precisely – not just abstractions and “lines-and-boxes”.

Specific Technologies:

  • Python Data Science Stack – NumPy, Pandas, Scikit Learn, various deep learning and NLP libraries including demonstrable experience with libraries relevant to data manipulation.
  • Langraph, LlamaIndex, and/or other agentic libraries.
  • Docker, Docker Compose, K8s, and related container orchestration technologies.
  • Hands‑on experience running local LLMs using technologies such as Ollama and Transformers.
  • Have a comfort level building and training and/or fine‑tuning models, not just calling cloud APIs, but you’re resourceful enough to recognize when you can get away with using cloud APIs.
  • Aren’t afraid of digesting academic papers, but you understand that it’s building things that matters.
  • Have an understanding of both the fundamentals and applications of different NLP approaches.

Nice-To-Haves:

  • Some familiarity with knowledge representation.
  • Previous experience with consulting on Defense projects involving software development, data science support, especially at service labs or joint commands.
About Jaxon: Jaxon is on a mission to make AI trustworthy.

Developed through R&D with the U.S. Department of Defense, Jaxon converts complex rule sets—whether from policy manuals, onboarding playbooks, compliance protocols, or SME conversations—into executable logic. It extracts the key ‘questions’ hidden in unstructured docs, builds rule‑driven evaluators, and tracks what’s being asked and why. The result: consistent, auditable, and provable reasoning—bringing structure, oversight, and trust to LLM‑driven workflows.

What Makes Jaxon Different?
  • Human First: Our leaders get it — life happens. Whether it’s flexible working hours or understanding the need for a mental health day, we prioritize your well‑being. Our CEO recognizes that happy people do the best work and he backs that up with an empathetically led culture — and I’m not just saying that because he’s going to read this post.
  • Laid‑Back Culture: We know that the best work comes from a place of comfort and authenticity. At Jaxon, you’re encouraged to be yourself, dress casually, and embrace your quirky side. It’s just one of the benefits of being an early‑stage, high‑growth startup.
  • Tech Lover’s Dream: From hackathons to tech talks with industry leaders, we fuel our passion for innovation with constant learning and development opportunities.
  • Grunt Work to Grand Projects: Every task you tackle, big or small, matters. You think I’m writing job descriptions for fun? Sometimes things need to get done and it’s an all hands on deck situation to get them to the finish line. We’re looking for team players who understand that achieving groundbreaking AI innovations sometimes involves rolling up your sleeves to tackle the essential, yet less glamorous tasks.
Added Perks:
  • Unlimited PTO: (seriously!)
  • Comprehensive benefits: Health, dental, and vision insurance 401k matching
  • Competitive Salary: We provide competitive compensation packages commensurate with experience
  • Stock Options: All compensation plans include stock options – join now and benefit from future acquisitions
  • Holidays Off: ALL federal holidays are paid time off.
About You:

You’re the type of person who’s naturally driven—not necessarily because you built a rocket in your backyard as a kid, but because you have a genuine curiosity about the world of technology. In your spare time, you might find yourself deep‑diving into technology subreddits, following the latest industry trends, or exploring which companies are innovating and how. Your passion isn’t just about staying informed; it’s about thinking how you can take these emerging concepts and scale them up, make them faster, or more impactful. At Jaxon, we’re looking for someone who sees a baseline and instinctively wants to build beyond it. If you’re ready to take what exists and make it extraordinary, you’ll fit right in.

We are looking for somebody who thrives in an environment where day‑to‑day priorities and tasks may rapidly change. The role is not a fit if you value “business as usual”.

We value team members who are bold enough to pitch creative ideas, even if they’re still rough around the edges or might not make the final cut.

Education and Length of Experience

Applicants should have a bachelor’s degree in Computer Science or similar. An MS or PhD in Computer Science, Statistics, Physics, Mathematics or any related Engineering field preferred. Candidates should have a minimum 5 years of experience in similar roles.

Jaxon is proud to be an equal opportunity workplace and is an affirmative action employer. We are committed to equal employment opportunities regardless of race, color, ancestry, religion, sex, sexual orientation, age, marital status, disability, or Veteran status.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Machine Learning Engineer
Machine Learning Engineer

Jaxon, Inc. • Lincoln (MA)

Hybrid
USD 100,000 - 130,000
Unlimited PTO
Competitive Salary
Stock Options
+1
Sales Associate - Startup AI Company
Sales Associate - Startup AI Company

Jaxon, Inc. • Boston (MA)

Hybrid
USD 45,000 - 60,000
Unlimited PTO
Health, dental, and vision insurance
401K matching
+3
Full-Stack Python Engineer
Full-Stack Python Engineer

Jaxon, Inc. • Lincoln (MA)

Hybrid
USD 80,000 - 120,000
Unlimited PTO
Competitive salary
Stock options
+1
Defense AI ML Engineer — NLP & LLMs (Remote)
Defense AI ML Engineer — NLP & LLMs (Remote)

Jaxon, Inc. • Boston (MA)

On-site
USD 115,000 - 145,000
Unlimited PTO
Comprehensive health benefits
401k matching
+2
Senior Machine Learning Scientist
Senior Machine Learning Scientist

Axon • Seattle (WA)

Hybrid
USD 290,000 - 501,000
Competitive salary
Discretionary paid time off
Paid parental leave
+4
Senior Machine Learning Engineer
Senior Machine Learning Engineer

Axon • Seattle (WA)

Hybrid
USD 150,750 - 221,000
Competitive salary and 401k with employer match
Discretionary paid time off
Paid parental leave
+3
Senior Machine Learning Scientist
Senior Machine Learning Scientist

National Society for Black Engineers • Seattle (WA)

On-site
USD 290,000 - 501,000
Competitive salary and 401(k) with employer match
Discretionary paid time off
Paid parental leave
+3
Mission Engineer - Overland Park, KS Kansas-Remote Kansas-Remote
Mission Engineer - Overland Park, KS Kansas-Remote Kansas-Remote

Axon Enterprise • Overland Park (KS), Northern (KY)

Hybrid
USD 120,000 - 160,000
Competitive salary discussed with 401k
Discretionary paid time off
Paid parental leave
+5
AI Evangelist - Program Manager
AI Evangelist - Program Manager

National Society for Black Engineers • Boston (MA)

On-site
USD 147,000 - 237,000
Mission Engineer - Colorado Springs Denver, Colorado, United States; Denver, Colorado, United S[...]
Mission Engineer - Colorado Springs Denver, Colorado, United States; Denver, Colorado, United S[...]

Axon Enterprise • Colorado Springs (CO), Northern (KY)

Hybrid
USD 128,000 - 204,000
401(k) match
Paid time off
Health, dental, vision
+2