Staff AI Engineer
HMH
full-time
Posted on:
Location Type: Office
Location: Pune • India
Visit company websiteExplore more
Job Level
About the role
- 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
- Use tools like MLflow and Docker to manage model lifecycle, reproducibility, and scalability
- Work closely with data engineers to ensure robust data pipelines and infrastructure for model training and inference
- Develop APIs and microservices to integrate AI models into enterprise applications and workflows
- Continuously monitor model performance and retrain as needed to maintain accuracy and relevance
- Ensure AI systems comply with enterprise security and data governance standards
Requirements
- 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
Benefits
- Health insurance
- Professional development opportunities
- Flexible work arrangements
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard skills
machine learningdeep learningPythonTensorFlowPyTorchScikit-learnlarge language modelsNLPcomputer visionRESTful APIs
Soft skills
clean codeefficient codereusable code