Position Name: Machine Learning Engineer

Location: Remote

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.

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: 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.

• Model Evaluation: Understanding of metrics and measurements of models to assess performance and apply rigorous evaluation practices to ensure reliability and effectiveness.

Experience with a majority of the following desired:

• Languages & Frameworks: Python (core and advanced), enterprise software development practices (modular design, testing frameworks such as pytest/unittest, CI/CD integration).

• Python Data Science Stack: NumPy, Pandas, Scikit-Learn, PyTorch, TensorFlow, Hugging Face Transformers; strong experience in data manipulation, feature engineering, and ML pipeline construction.

• LLM/Agentic Frameworks: LangChain, Langraph, LlamaIndex, and related libraries for building AI-assisted applications.

• Systems & Infrastructure: Docker, Docker Compose, Kubernetes, container orchestration, cloud deployment (AWS, GCP, or Azure), monitoring/logging frameworks (Prometheus, ELK, etc.).

• MLOps Practices: Building reproducible ML pipelines, model packaging/deployment, version control for ML artifacts (e.g., MLflow, DVC), integration with CI/CD workflows.

• Modeling Expertise: Hands-on experience running local LLMs using Ollama and Hugging Face Transformers; building, training, and fine-tuning models for enterprise use cases.

• NLP Competency: Strong foundation in both fundamentals (tokenization, embeddings, sequence modeling) and applied approaches (question answering, summarization, RAG, fine-tuning).

• Software Delivery: Delivery of ML or software systems in Defense or other highly regulated domains, including direct interaction with government customers (e.g., service labs, 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 Software Engineering with a minimum 5 years of experience in applicable 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.