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!