Staff Applied Machine Learning Engineer

About the role

Legion was built to move beyond AI hype—delivering secure, reliable systems that work alongside the people tackling the world’s most critical challenges. Born from a Department of Defense partnership and trusted by leaders across government and enterprise, Legion embeds intelligence inside complex systems, unlocking data, accelerating human workflows, and strengthening mission-critical systems. We don’t replace workflows—we optimize them, ensuring quality, efficiency, and reliability inside the platforms our partners already use. With world-class collaborators like Palantir, Nvidia, HPE, and Oracle, we’re building intelligent infrastructure that enhances human capability and drives impact at the edge and across a range of enterprises.

Responsibilities

• Lead the development of machine learning models and data-driven algorithms for high impact projects
• Lead the design and development of complex systems involving knowledge graphs and advanced entity detection
• Collaborate with product and platform teams to own ML solutions end-to-end
• Understand the runtime complexity of algorithms and the cost to run ML models at a production scale
• Clearly communicate modeling decisions, tradeoffs, and limitations to technical and non-technical stakeholders
• Build, deliver, and maintain enterprise products, ensuring they meet high-quality standards while shipping fast
• Take ownership of your work, from design to implementation and maintenance, and drive projects to successful completion
• Support production systems, handling debugging challenges in distributed systems to ensure reliability and uptime
• Adapt quickly, bringing in latest developments in AI and machine learning, and proactively apply this knowledge to drive innovation within the company

Requirements

• US Citizenship or Greencard is required due to US Government contract requirements
• Bachelor’s or Master’s degree in Computer Science, Machine Learning, NLP. PhD is preferred
• 10+ years of experience in ML engineering with a proven track record of building models for large-scale systems
• Proven experience designing, building, evaluating, and optimizing Agentic AI systems, including reasoning loops, tool use, orchestration patterns, context management
• Strong communication skills and the ability to collaborate effectively with cross-functional teams
• Comfortable supporting production systems and debugging challenges in distributed systems
• Demonstrated effectiveness in using AI tooling to accelerate research and development workflows
• Growth mindset and low ego- you’re eager to pick up new tools and technologies, learn from others, and being open to changing course when it’s right
• Drive to foster a culture of collaboration, psychological safety, and technical excellence
• Strong advocate for inclusion in technical discussions at all levels

Qualifications

• US Citizenship or Greencard is required due to US Government contract requirements
• Bachelor’s or Master’s degree in Computer Science, Machine Learning, NLP. PhD is preferred
• 10+ years of experience in ML engineering with a proven track record of building models for large-scale systems
• Proven experience designing, building, evaluating, and optimizing Agentic AI systems, including reasoning loops, tool use, orchestration patterns, context management
• Strong communication skills and the ability to collaborate effectively with cross-functional teams
• Comfortable supporting production systems and debugging challenges in distributed systems
• Demonstrated effectiveness in using AI tooling to accelerate research and development workflows
• Growth mindset and low ego- you’re eager to pick up new tools and technologies, learn from others, and being open to changing course when it’s right
• Drive to foster a culture of collaboration, psychological safety, and technical excellence
• Strong advocate for inclusion in technical discussions at all levels

Skills

• Start-up experience or comfort in 0→1 product environments
• Familiarity with Kubernetes is a plus
• Experience integrating machine learning models and data driven algorithms into larger system architectures

Benefits

• No benefits mentioned

Pay

$230,000 - $270,000 USD

Schedule

Remote USA