Junior Machine Learning Engineer

Junior ML Engineer

📍 Indore, Madhya Pradesh
🕒 Full-Time | Onsite
📅 Monday–Friday (5 Days)

Want to impact the lives of 10 million students across Bharat?

At Arivihan, we're building an AI Tutor that doesn't just answer questions it teaches. By combining personalization, scalability, and affordability, we're creating learning experiences that can reach every student, regardless of geography or background.

We've already transformed outcomes for students preparing for State Board exams, starting with the MP Board. Now, we're expanding across multiple state boards and competitive examinations with a vision to become India's most trusted AI-powered learning platform.

If you're excited about solving real-world problems with AI, shipping products that impact millions, and growing alongside a fast-moving team, we'd love to meet you.

About the Role
We are looking for a Junior ML Engineer who is passionate about Generative AI and eager to build production-ready AI features that directly improve learning outcomes for students.
You'll work closely with senior engineers and product teams to design, develop, test, and deploy AI-powered experiences using LLMs, NLP techniques, and modern machine learning workflows.

Key Responsibilities

• Collaborate with Senior ML Engineers to build, test, and deploy AI-powered product features
• Develop clean, scalable Python code and reusable modules
• Implement and optimize LLM workflows, including prompting, tool/function calling, and retrieval-based systems (RAG)
• Integrate and maintain LLM APIs such as OpenAI, Gemini, and other foundation models
• Improve output quality through prompt engineering, evaluation frameworks, and structured testing
• Design and execute regression testing, validation pipelines, and quality checks
• Build and consume REST APIs to integrate AI capabilities into backend systems
• Document prompts, experiments, datasets, and feature logic for maintainability and scalability
• Work closely with Product, Engineering, and Content teams to deliver impactful learning experiences

What We're Looking For

• 1+ years of hands-on experience building AI or Machine Learning solutions
• Practical experience with Generative AI and Large Language Models (LLMs)
• Experience with prompting, tool/function calling, retrieval-based systems (RAG), and model evaluation
• Strong Python programming skills
• Understanding of LLM APIs, inference workflows, output validation, and prompt iteration
• Experience building or integrating REST APIs
• Familiarity with PyTorch and/or TensorFlow
• Understanding of NLP concepts, tokenization, prompting strategies, and evaluation methodologies
• Strong problem-solving ability and a continuous learning mindset
• Excellent collaboration and communication skills

Why Join Arivihan?

• Build AI products that impact millions of learners across India
• Work on cutting-edge Generative AI and LLM applications
• Collaborate with a highly driven product and engineering team
• Gain ownership from day one and grow rapidly in a high-impact environment
• Help shape the future of education through technology

📩 Apply Now
Share your resume at [email protected]
Subject: Junior ML Engineer | Arivihan