HMH
Staff AI Engineer
🇮🇳 Pune, Maharashtra
🕑 Full-Time
💰 TBD
💻 Software Engineering
🗓️ November 9th, 2025
Edtech.com's Summary
- Design, train, and deploy machine learning and deep learning models using frameworks like TensorFlow, PyTorch, and Scikit-learn.
- Implement and fine-tune large language models (LLMs) using tools such as LangChain, RAG, and vector databases; experience with GPTs, LLaMA preferred.
- Utilize MLOps tools like MLflow and Docker to manage model lifecycle and support GenAI systems.
- Collaborate with data engineers to ensure robust data pipelines and infrastructure for model training and inference.
- Develop APIs and microservices for AI model integration into enterprise workflows.
- Monitor model performance continuously and retrain models to maintain accuracy.
- Ensure AI systems comply with enterprise security and data governance standards.
- Required skills: 8–12 years AI/ML engineering experience, proficiency in Python, experience with AI frameworks (TensorFlow, PyTorch), cloud platforms (AWS, Azure, GCP), MLOps tools, RESTful APIs and microservices architecture.
- Preferred experience includes exposure to educational technology, enterprise data environments, transactional AI integration, and familiarity with data warehouses and governance frameworks.
- Employer: HMH Technology India Pvt. Ltd., the technology and innovation arm of HMH focused on global AI product development.
Staff AI Engineer Full Description
HMH is a learning technology company committed to delivering connected solutions that engage learners, empower educators and improve student outcomes. As a leading provider of K-12 core curriculum, supplemental and intervention solutions, and professional learning services, HMH partners with educators and school districts to uncover solutions that unlock students' potential and extend teachers' capabilities.
HMH serves more than 50 million students and 4 million educators in 150 countries. HMH Technology India Pvt. Ltd. is our technology and innovation arm in India focused on developing novel products and solutions using cutting-edge technology to better serve our clients globally. HMH aims to help employees grow as people, and not just as professionals. For more information, visit www.hmhco.com
We are seeking a mid-level AI Engineer to join our team focused on building scalable, production-ready AI solutions that support analytics, automation, and intelligent decision-making. This role will contribute to the development and deployment of machine learning models, generative AI systems, and AI-enhanced data workflows across platforms.
Key Responsibilities
- Model Development & Deployment: Design, train, and deploy machine learning and deep learning models using frameworks like TensorFlow, PyTorch, and Scikit-learn.
- Generative AI & LLMs: Implement and fine-tune large language models (LLMs) using tools such as LangChain, RAG, and vector databases. Experience with GPTs, LLaMA, or similar models is preferred
- MLOps & GenAIOps: Use tools like MLflow and Docker to manage model lifecycle, reproducibility, and scalability. Support production-grade GenAI systems
- Data Engineering Collaboration: Work closely with data engineers to ensure robust data pipelines and infrastructure for model training and inference.
- Integration & APIs: Develop APIs and microservices to integrate AI models into enterprise applications and workflows.
- Monitoring & Optimization: Continuously monitor model performance and retrain as needed to maintain accuracy and relevance.
- Security & Governance: Ensure AI systems comply with enterprise security and data governance standards.
Skills & Qualifications
- 8-12 years of experience in AI/ML engineering or related roles.
- Proficiency in Python and experience with AI frameworks (TensorFlow, PyTorch).
- Familiarity with cloud platforms (AWS, Azure, GCP) for model deployment.
- Experience with MLOps tools (MLflow, Docker) and GenAI deployment.
- Strong understanding of LLMs, NLP, and computer vision techniques.
- Ability to write clean, efficient, and reusable code.
- Experience with RESTful APIs and microservices architecture.
Preferred Experience
- Exposure to educational technology or enterprise data environments.
- Experience integrating AI into transactional systems.
- Familiarity with data warehouses and data governance frameworks.
HMH Technology Private Limited is an Equal Opportunity Employer and considers applicants for all positions without regard to race, colour, religion or belief, sex, age, national origin, citizenship status, marital status, military/veteran status, genetic information, sexual orientation, gender identity, physical or mental disability or any other characteristic protected by applicable laws. We are committed to creating a dynamic work environment that values diversity and inclusion, respect and integrity, customer focus, and innovation. For more information, visit https://careers.hmhco.com/. Follow us on Twitter, Facebook, LinkedIn, and YouTube.
