Machine Learning Engineer

About The Company IM is a pioneering organization dedicated to transforming education through innovative technology solutions. Our mission is to develop intelligent learning platforms that personalize education, making it accessible, engaging, and effective for students across diverse backgrounds. With a focus on leveraging cutting-edge artificial intelligence and machine learning techniques, IM aims to create impactful tools that facilitate meaningful learning experiences. Our team comprises passionate professionals committed to advancing educational equity and excellence, and we are continuously pushing the boundaries of what technology can achieve in the realm of education. About The Role We are seeking a highly skilled Machine Learning Engineer to lead the development of our Adaptive Learning Engine based in Bangalore. This on-site role offers an exciting opportunity to build a core component of our educational platform that personalizes learning paths for students preparing for JEE/NEET exams. The successful candidate will be responsible for designing, implementing, and deploying a sophisticated knowledge tracing model that can accurately infer students' misconceptions and adapt content in real-time. You will work closely with data scientists, product managers, and engineers to create an end-to-end solution that enhances student engagement and learning outcomes. The project timeline is set for 12 weeks to reach Release 1.0, and we are looking for an engineer capable of building this adaptive core from scratch. This role requires a combination of strong technical expertise, innovative thinking, and a passion for educational technology. If you thrive in a fast-paced environment and are eager to make a tangible impact on learners' lives, this opportunity is ideal for you. Qualifications The ideal candidate will possess a solid foundation in machine learning, with practical experience in developing and deploying models in real-world scenarios. You should have a proven track record of implementing attention mechanisms, sequence models, and embeddings from scratch, rather than relying solely on pre-built libraries. Proficiency in Python and PyTorch (or TensorFlow) is essential, along with familiarity with standard data science stacks such as NumPy, pandas, and scikit-learn. You must have experience training end-to-end models on real datasets and deploying them in production environments where they are actively used. The ability to critically analyze research papers and discern valuable ideas from noise is crucial, as is the capacity to justify modeling decisions with sound reasoning beyond mere improvements in validation metrics. Strong problem-solving skills, curiosity, and a passion for innovation are highly valued. Preferred qualifications include advanced coursework or experience in natural language processing, sequence modeling, or related fields, as well as experience with knowledge tracing, IRT, or educational data mining. Experience working with complex telemetry data and participating in competitive data science challenges like Kaggle will be considered a plus. Responsibilities Design, develop, and implement a knowledge tracing model using benchmarks such as BKT, DKT, DKVMN, or AKT, selecting the most effective approach through rigorous experimentation and evaluation. Build a misconception inference engine that predicts specific student misunderstandings based on their incorrect answers, mapping to a comprehensive taxonomy of misconceptions. Develop cold-start strategies to effectively personalize learning experiences for new students with no prior history. Extract behavioral signals from interaction telemetry to infer engagement levels and confidence without explicit input, enhancing model accuracy and responsiveness. Create a real-time inference pipeline capable of serving predictions seamlessly to web and mobile clients, ensuring low latency and high reliability. Establish an evaluation framework to continuously monitor and improve model performance, ensuring the adaptive engine evolves effectively over time. Collaborate with cross-functional teams to integrate the adaptive engine into the broader platform, aligning technical development with educational objectives. Benefits At IM, we offer a competitive salary package aligned with industry standards, along with ESOP options that allow you to share in the company's growth and success. Our work environment fosters innovation, collaboration, and continuous learning, providing opportunities for professional development and skill enhancement. The role includes comprehensive health benefits, a supportive team culture, and the chance to contribute to a meaningful mission of transforming education through technology. We also prioritize work-life balance and a dynamic, inclusive workplace where diverse perspectives are valued. Equal Opportunity IM is an equal opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all employees. We do not discriminate based on race, religion, gender, sexual orientation, age, disability, or any other protected characteristic. Our hiring decisions are based solely on qualifications, skills, and experience, ensuring a fair and equitable process for all applicants.