HMH logo

HMH

Staff AI Engineer

🇮🇳 Pune, Maharashtra 🕑 Full-Time 💰 TBD 💻 Software Engineering 🗓️ February 8th, 2026
LMS Python Docker

Edtech.com's Summary

HMH is hiring a Staff AI Engineer to develop and deploy scalable, production-ready AI solutions that enhance analytics, automation, and intelligent decision-making. The role involves designing machine learning and deep learning models, implementing generative AI systems, collaborating with data engineers, and integrating AI models via APIs into enterprise applications.

Highlights
  • Design, train, and deploy ML and DL models using TensorFlow, PyTorch, and Scikit-learn.
  • Implement and fine-tune large language models (LLMs) using LangChain, RAG, and vector databases, with experience preferred in GPTs or LLaMA.
  • Use MLOps tools such as MLflow and Docker for managing model lifecycle and production-grade GenAI systems.
  • Collaborate with data engineers to maintain robust data pipelines and infrastructure.
  • Develop APIs and microservices for AI integration into enterprise workflows.
  • Monitor and optimize model performance to maintain accuracy and relevance.
  • Ensure AI systems meet enterprise security and data governance standards.
  • Require 8-12 years of AI/ML engineering experience, proficiency in Python and AI frameworks, and familiarity with cloud platforms (AWS, Azure, GCP).
  • Preferred experience includes exposure to educational technology, enterprise data environments, and AI integration in transactional systems.
  • HMH serves over 50 million students and 4 million educators worldwide and focuses on K–12 educational technology solutions.

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.