Senior Machine Learning Engineer

Jaxon.AI is focused on making AI trustworthy for mission-critical environments, particularly within the Department of Defense. The Senior Machine Learning Engineer will advance machine learning capabilities to enhance the reliability and security of AI applications in national security operations.

Responsibilities:

• 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
• 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'

Requirements:

• Bachelor's degree in Computer Science or similar
• Minimum 5 years of experience in similar roles
• Solid machine learning engineering experience, particularly with NLP applications and unstructured data in a cloud environment
• Fluency in Python
• Intimately familiar with the model deployment process, optimization of LLM parameters for specific behaviors, and a general understanding of LLM functionality and use cases
• 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'
• Hands-on experience running local LLMs using technologies such as Ollama and Transformers
• Comfort level building and training and/or fine-tuning models, not just calling cloud APIs, but resourceful enough to recognize when to use cloud APIs
• Understanding of both the fundamentals and applications of different NLP approaches
• MS or PhD in Computer Science, Statistics, Physics, Mathematics or any related Engineering field
• Previous experience with AI/LLM governance and guardrail development
• 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