AI Engineer

Viraaj HR Solutions Private Limited; Ahmedabad District; On-site; INR 1,800,000 - 3,000,000; Full time

Benefits offered by this job

On-site role in India

Job summary

Viraaj HR Solutions Private Limited seeks an experienced AI/ML Engineer to design, train, and deploy scalable models for enterprise use. You will build AI pipelines, collaborate with data scientists and software engineers, and productionize models on cloud platforms.

You will integrate models into enterprise applications via REST APIs, monitor drift, and contribute to explainability and fairness initiatives while writing clean code and documenting AI systems for scalability.

Qualifications

• Must-have: Python, TensorFlow, PyTorch, scikit-learn, MLflow, Docker, Git, REST API.
• Preferred: LangChain, FastAPI, Azure ML.

Responsibilities

• Design, train, and deploy scalable ML models for production environments.
• Develop AI-powered pipelines for classification, NLP, or recommendations.
• Collaborate with data scientists and engineers to productionize models on cloud platforms (AWS/Azure/GCP).
• Integrate AI models into enterprise apps via REST/gRPC APIs for low-latency performance.
• Monitor model drift and retraining workflows; support explainability and fairness initiatives.
• Write clean, maintainable code and document AI systems for scalability and compliance.

Skills

Python

TensorFlow

PyTorch

scikit-learn

MLflow

Tools

Docker

Git

REST API

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