Industry: Education 

About the Role 

We are looking for an ML Engineer to join our customer’s team and take ownership of improving and maintaining their free-text assistant system. This position combines machine learning expertise (especially with large language models) with strong data engineering and software engineering skills. 

You will work on a unique AI-driven education platform that helps educators evaluate and provide feedback on open-ended assignments more efficiently, while providing students with personalized and adaptive learning experiences. 

The ideal candidate is a hands-on engineer who can bridge the gap between data science, data engineering, and software engineering, driving both experimentation and production-ready solutions. 

What you’ll do 

• Analyze and understand the current system architecture, data pipelines, and limitations. 
• Maintain and improve the free-text assistant: identify issues, propose and implement solutions. 
• Build and optimize data workflows to support ML/LLM tasks (data ingestion, preprocessing, monitoring). 
• Experiment with ML/LLM techniques to improve scalability, performance, and accuracy. 
• Collaborate with product and engineering teams to align improvements with business needs. 
• Participate in technical discussions and help shape the roadmap. 
• (Optional) Contribute to API/service integration if you have web framework experience. 

First tasks will include: 

• Deep dive into the system to fully understand its components and data flows. 
• Assess product and business needs to align them with technical improvements. 
• Contribute to ongoing work on progressive model development. 

What we’re looking for 

• Strong programming skills in Python. 
• Hands-on experience with LLMs (e.g., LangChain or similar frameworks). 
• Practical ML expertise (e.g., AutoGluon or equivalent libraries). 
• Proven experience in data engineering — pipelines, structured/unstructured data. 
• Solid software engineering background, including integrating ML into real-world systems. 
• Strong analytical and critical thinking skills. 

Nice to have 

• Experience with FastAPI or other modern Python frameworks. 
• Research background or experience experimenting with new ML/LLM methods. 

Who you are 

• Critical thinker — you evaluate systems holistically. 
• Problem solver — you enjoy both experimenting and shipping solutions. 
• Curious & proactive — you explore new approaches and share ideas. 
• Collaborative — you thrive in discussions, challenge assumptions, and align with business needs. 

Why you’ll enjoy working with us 

• Work on an innovative AI platform shaping the future of higher education. 
• Be part of a team that values initiative, ownership, and continuous learning. 
• Collaborate with top professionals at the intersection of AI, education, and product innovation. 
• Competitive salary, flexibility, and an opportunity to make a real impact. 

Let's Talk!