Machine Learning Engineer, Memory

Requirements

• PhD in Computer Science, Mathematics or related field, or 3+ years of relevant industry experience
• Experience creating machine learning algorithms for vision and/or language problems and deploying them as production-level systems
• Expert, hands-on knowledge of:
  • Foundational models and vision and/or language, incl. local deployment and model post-training of open weights models
  • Memory, state, or long-horizon behaviour in foundation models: your experience may come from language agents and long-context methods, reinforcement learning under partial observability, world models, or sequence modelling
  • Python and associated ML tools/frameworks (numpy, scipy, sklearn, pytorch)
• Experience with working collaboratively with other developers, following a flexible and agile approach

What You'll Be Doing

• Design new memory architectures that work across problems, modalities and products
• Define new ways to measure and evaluate memory performance
• Help create and deliver production-ready, scalable and high-quality learning models and associated algorithms
• Critically assess the effectiveness of such models and make recommendations for the ongoing roadmap

Perks and Benefits

• We pay 100% for benefits except for PMI (for dependents)
• Benefits package includes pension, private medical insurance, health care cash plan, dental insurance, disability and life insurance, critical illness, cycle to work scheme, flu shots, health checks, and meals
• Robust mental well-being program through Modern Health, providing free therapy and coaching for employees & dependents