Senior AI Engineer

PierianDx India Private Limited; Pune District; On-site; INR 800,000 - 1,200,000; Full time

Job summary

PierianDx India Private Limited in Pune is seeking an AI/ML Engineer to train and deploy Large Language Models (LLMs) for effective real-world problem-solving. You will design and optimize AI/ML pipelines, ensuring model performance and scalability.

The ideal candidate has strong experience in deploying LLMs using frameworks like PyTorch or TensorFlow, along with proficiency in Python and cloud infrastructure. Join us to drive innovation in AI and deepen your expertise in machine learning.

Qualifications

• Experience in training and deploying LLMs using PyTorch, TensorFlow, or Hugging Face Transformers.
• Strong programming skills in Python and familiarity with data manipulation libraries.
• Hands-on experience in developing AI/ML pipelines and model optimization.

Responsibilities

• Train, fine-tune, and deploy LLMs for real-world problems.
• Design and optimize AI/ML pipelines for model development.
• Collaborate with teams for AI solution integration.

Skills

LLM training and deployment

AI/ML pipeline optimization

Programming in Python

Tokenization and prompt engineering

Building RAG pipelines

Cloud infrastructure (AWS/GCP/Azure)

MLOps best practices

Problem-solving

Tools

PyTorch

TensorFlow

Docker

Kubernetes

NumPy

Pandas

Scikit-learn

Vector databases (FAISS, Pinecone)

Job description

Responsibilities

• Train, fine-tune, and deploy Large Language Models (LLMs) to solve real-world problems effectively.
• Design, implement, and optimize AI/ML pipelines to support model development, evaluation, and deployment.
• Collaborate with Architect, software engineers, and product teams to integrate AI solutions into applications.
• Ensure model performance, scalability, and efficiency through continuous experimentation and improvements.
• Work on LLM optimization techniques, including Retrieval-Augmented Generation (RAG), prompt tuning, etc.
• Manage and automate the infrastructure necessary for AI/ML workloads while keeping the focus on model development.
• Work with DevOps teams to ensure smooth deployment and monitoring of AI models in production.
• Stay updated on the latest advancements in AI, LLMs, and deep learning to drive innovation.

Qualifications

• Strong experience in training, fine-tuning, and deploying LLMs using frameworks like PyTorch, TensorFlow, or Hugging Face Transformers.
• Hands‑on experience in developing and optimizing AI/ML pipelines, from data preprocessing to model inference.
• Solid programming skills in Python and familiarity with libraries like NumPy, Pandas, and Scikit‑learn.
• Strong understanding of tokenization, embeddings, and prompt engineering for LLM‑based applications.
• Hands‑on experience in building and optimizing RAG pipelines using vector databases (FAISS, Pinecone, Weaviate, or ChromaDB).
• Experience with cloud‑based AI infrastructure (AWS, GCP, or Azure) and containerization technologies (Docker, Kubernetes).
• Experience in model monitoring, A/B testing, and performance optimization in a production environment.
• Familiarity with MLOps best practices and tools (Kubeflow, MLflow, or similar).
• Ability to balance hands‑on AI development with necessary infrastructure management.
• Strong problem‑solving skills, teamwork, and a passion for building AI‑driven solutions.

Velsera is an Equal Opportunity Employer: Velsera is proud to be an equal opportunity employer committed to providing employment opportunity regardless of sex, race, creed, colour, gender, religion, marital status, domestic partner status, age, national origin or ancestry.