← All roles

Staff Machine Learning Engineer

Themis AI· Themis AI
Apply on company site
Hybrid (Remote / Cambridge, MA)full-timePosted 10mo ago

About the role

Staff Machine Learning Engineer to lead development of high-performance Python-based ML libraries, translating research into scalable APIs and systems for uncertainty-aware AI infrastructure.

Responsibilities

  • Support the design, development, optimization and testing of our core ML libraries
  • Collaborate closely with AI research teams to translate new innovations into high-performance, easy-to-use APIs and scalable systems
  • Contribute to architectural decisions and long-term technical roadmaps
  • Drive strong engineering practices: code reviews, automated testing, CI/CD, docs, etc.
  • Mentor engineers and help maintain a culture of technical rigor and mission alignment

Qualifications

  • 5+ years of software engineering experience in Python and ML model development
  • Deep understanding of machine learning fundamentals, particularly neural networks, optimization, probability theory, and linear algebra
  • Strong software engineering practices: Git-based workflows, pull request reviews, testing infrastructure, release processes
  • Major contributions to production-grade AI/ML frameworks, toolkits, or open-source tools
  • Clear communicator with the ability to operate autonomously and prioritize effectively in a startup environment
  • Experience working in high-performance, collaborative software teams
  • Mission-driven mindset: you care deeply about building systems that matter

Skills

Required:PythonMachine Learning fundamentalsSoftware engineering practices
Last updated Sep 30, 2026. Interested? Apply on the company site.

About Themis AI

Make any AI model safe and reliable.

Themis AI | Make any AI model safe and reliable.

uncertainty quantificationmachine learningAI safety

More open roles at Themis AI