Staff Applied Machine Learning Engineer

Staff Applied Machine Learning Engineer (Remote USA)

***US Citizenship or Greencard is required due to US Government contract requirements***

Job Summary:

In this role, you will leverage your expertise in AI/ML engineering to design, develop, and deploy innovative machine learning and algorithmic solutions. 

If you are adept at building models that solve hard problems, we encourage you to apply. You will collaborate closely with platform engineers and product partners, bringing a strong product orientation to your ML work.

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

Required Skills and 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
• Experience building and supporting knowledge graph and semantic representation 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

Preferred Qualifications:

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

Equal Opportunity Statement

Legion provides equal employment opportunities to all employees and applicants for employment and prohibits discrimination and harassment of any type without regard to race, color, religion, age, sex, national origin, disability status, genetics, protected veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by federal, state or local laws.