Machine Learning Engineer - AI Company
- Discuss this with your agent
- Open in Claude
- Open in ChatGPT
Role details
Tech stack
+9 more
Job description
- Collaborate Across Teams: Work with cross-functional, geographically distributed teams to integrate ML models into our existing systems and workflows, enhancing product capabilities.
Requirements
Do you have experience in Technical Proficiency?, Do you have a Bachelorâs degree?, * 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
- 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., 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., * Bachelorâs (Required)
Experience:
- Machine learning: 5 years (Required)
- Python: 5 years (Required)
Benefits & conditions
Pulled from the full job description
- Health insurance
- 401(k) matching
- Paid time off
- Vision insurance
- Dental insurance
- Stock options
- Unlimited paid time off, * 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., * 401(k) matching
- Dental insurance
- Flexible schedule
- Health insurance
- Paid time off
- Vision insurance
Application Question(s):
- Will you now or in the future require sponsorship for employment visa status (e.g., H-1B visa status)?
- Are you a U.S. citizen? This is required due to the nature of our work with the U.S. Government.
- Please provide your Linkedin URL.
About the company
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.
Apply for this position
This job is hosted externally. Click below to view the full posting and apply.
Apply on indeed.comGood distractions
Talks and stories from around this role â technically off-topic, practically not.
Moments
Explore playlistsVideos
See allRelated articles
See all
How to Become an AI Engineer
MLOps And AI Driven Development
MLOps â Whatâs the deal behind it?
Highest Paying Tech Companies for Developers