ML/AI Engineer

About the role

We are seeking an experienced ML/AI Engineer to design and implement advanced machine learning models and innovative math teaching techniques that enhance learner comprehension and progression. This position sits at the intersection of artificial intelligence and education, with a strong emphasis on delivering measurable product impact.

Key responsibilities

• Design and develop ML/AI systems that support personalized mathematics learning.
• Research and implement teaching techniques that improve comprehension and retention of mathematical concepts.
• Build and evaluate computer‑vision or multimodal features for educational workflows.
• Collaborate with curriculum, product, and engineering teams to translate teaching ideas into product features.
• Analyze learner outcomes and iterate on models, prompts, and instructional methods.
• Document experiments, model behavior, and best practices for safe and reliable deployment.

Required profile

• 6+ years of relevant experience in machine learning or AI.
• Strong background in computer vision and deep learning.
• Proven experience building and evaluating production‑grade ML systems.
• Understanding of educational technology, learning science, or instructional design.
• Ability to translate research ideas into practical product improvements.

Required skills

• Python programming.
• Deep learning frameworks such as PyTorch or TensorFlow.
• Computer vision techniques.
• Deep learning model development.
• Production‑grade ML system engineering.