Machine Learning Engineer - AI Startup

About The Job

At Jaxon, we are on a mission to make AI trustworthy. The adoption of large language models (LLMs) in corporate settings is hindered by the “hallucination problem,” where AI models produce inaccurate or entirely fabricated responses. For regulated industries and high-risk applications, this issue is preventing LLM use altogether. This role is crucial in overcoming these challenges.

As a Machine Learning Engineer at Jaxon, you will be responsible for developing and implementing machine learning models and algorithms to create more reliable AI systems. You will play a key role in delivering complex AI-driven solutions for customers, with a particular focus on defense applications as Jaxon is dual-use.

We have a patent pending product, Domain-Specific AI Logic (DSAIL), that forms the basis of most of our deployments. This role will be focused on increasing security and reliability for our customers through implementing DSAIL and building on its capabilities. A particular emphasis is placed on addressing the fundamental limitations of LLMs – they don’t understand context and logical reasoning. Once an LLM deviates from the right path, errors are more likely to occur. Training an AI away from this type of yes/no logic and into more human-like reasoning is at the core of this position.

Responsibilities

• Collaborate Across Teams: Work with cross-functional 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: Demonstrate solid machine learning engineering experience, particularly with NLP applications and unstructured data in a cloud environment. Fluency in Python is essential.
• Understand Machine Learning Models: Needs to be intimately familiar with the process, but deploying ML models will not be a regular part of the job.

Soft Skills

• Able to context switch as needed
• Able to compartmentalize different projects with different goals
• Able to communicate clearly and precisely - not just abstractions and "lines-and-boxes"

Specific Technologies

• Docker, Docker Compose
• Python Data Science Stack - NumPy, Pandas, Scikit Learn, Various deep learning packages
• Familiarity with calling web API's and making web API's
• Familiarity with modular architectures - IPC, publish-subscribe, etc…

Nice-To-Haves

• Familiarity with predicate logic
• Familiarity with Extended Backus-Naur Form (EBNF) notation for language grammars
• Lisp languages

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.

Education And Length Of Experience

Preferably applicants will have a bachelor's degree in Computer Science or similar, however at Jaxon, we value skills and dedication over degrees. We strongly encourage people without degrees to apply if you have at least 5 years of experience.

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.
• 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.
• 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. 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
• 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. Yes, even Indigenous People’s Day

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.