AI Engineer

Viraaj HR Solutions Private Limited; Bengaluru; On-site; INR 2,500,000 - 4,500,000; Full time

Job summary

Viraaj HR Solutions Private Limited in Bengaluru is seeking an experienced ML engineer to design, train, and deploy scalable AI models for production-grade applications. You will work across the full AI lifecycle from data ingestion to deployment, with a focus on performance and reliability.

Collaborate with data scientists and software engineers to productionize models on cloud platforms, integrate via REST/gRPC, monitor drift, and contribute to model explainability and governance.

Qualifications

• Must-have proficiency in Python and ML frameworks (TensorFlow, PyTorch, scikit-learn).
• Experience deploying ML models with MLflow and REST APIs.
• Hands-on experience with Docker, Git, and cloud platforms (AWS/Azure/GCP).

Responsibilities

• Design, train, and deploy scalable ML models for production environments.
• Develop AI pipelines for classification, regression, NLP, or recommendations.
• Collaborate with data scientists and engineers to productionize models on cloud.
• Integrate models into enterprise apps via REST/gRPC APIs.
• Monitor drift, retraining workflows, and contribute to fairness initiatives.
• Write clean code and document AI systems for scalability and compliance.

Skills

Python

TensorFlow

PyTorch

scikit-learn

MLflow

REST API

Tools

Docker

Git

LangChain

FastAPI

Azure ML

Job description

About The Opportunity

A fast-growing technology firm in the AI & SaaS sector, we build and deploy enterprise-grade machine learning and generative AI solutions that automate workflows, enhance customer intelligence, and drive decision‑making at scale. Our engineers work across the full AI lifecycle—from data ingestion and model training to deployment, monitoring, and continuous improvement—using modern tooling and cloud‑native architectures to deliver real business value.

Role & Responsibilities

• Design, train, and deploy scalable ML models using frameworks like TensorFlow, PyTorch, and scikit‑learn for production environments.
• Develop and optimize AI-powered pipelines for classification, regression, NLP, or recommendation systems based on business use cases.
• Collaborate with data scientists and software engineers to productionize models using MLOps practices on cloud platforms (AWS/Azure/GCP).
• Integrate AI models into enterprise applications via REST/gRPC APIs and ensure low-latency, high-availability performance.
• Monitor model drift, implement retraining workflows, and contribute to model explainability and fairness initiatives.
• Write clean, maintainable code and document AI systems for scalability, compliance, and team knowledge sharing.

Skills & Qualifications

• Must-Have
• Python
• TensorFlow
• PyTorch
• scikit‑learn
• MLflow
• Docker
• Git
• REST API
• Preferred
• LangChain
• FastAPI
• Azure ML

Benefits & Culture Highlights

• Opportunity to own high-impact AI projects from concept to production.
• Collaborative, fast-paced engineering culture with mentorship from senior AI leads.
• On-site role in India with modern infrastructure and innovation-focused work environment.

Skills: scikit-learn,ml,python,docker,git,tensorflow,pytorch