Ai Ml Engineer

TRUSTKLUB Consulting; Navi Mumbai; On-site; INR 1,500,000 - 2,500,000; Full time

Job summary

TRUSTKLUB Consulting is seeking an AI/ML professional with 4–7 years of experience to design, develop and deploy large language models using PyTorch, TensorFlow, and related frameworks. The role includes building generative AI models for text, image and video tasks, applying prompt engineering, and collaborating with cross-functional teams to integrate models into larger systems.

The ideal candidate should have a strong foundation in DL, LLMs, embeddings, and model serving, with hands-on Python

Qualifications

• Bachelor's degree in CSE or equivalent from a reputed institution.
• 4-7 years of experience in AI/ML with deep learning focus.
• Strong understanding of LLMs, embeddings, and model serving.

Responsibilities

• Design, develop, and deploy large language models using PyTorch, TensorFlow, or other relevant frameworks.
• Develop generative AI models for text generation, image synthesis, or video creation.
• Implement prompt engineering techniques to improve model performance on specific tasks.
• Collaborate with cross-functional teams to integrate ML models into larger systems.
• Troubleshoot issues related to model serving, fine-tuning, and deployment.

Skills

Deep Learning

PyTorch

TensorFlow

Python

LLMs

Embeddings

Vector Databases

Model Serving

Education

Bachelor's degree in Computer Science and Engineering (CSE) or equivalent

Tools

TorchServe

Triton Inference Server

KFServing

Job description

Roles and Responsibilities

• Design, develop, and deploy large language models using PyTorch, TensorFlow, or other relevant frameworks.


• Develop generative AI models for text generation, image synthesis, or video creation.


• Implement prompt engineering techniques to improve model performance on specific tasks.


• Collaborate with cross-functional teams to integrate ML models into larger systems.


• Troubleshoot issues related to model serving, fine-tuning, and deployment.

Desired Candidate Profile

• 4-7 years of experience in Artificial Intelligence/Machine Learning domain with expertise in Deep Learning.


• Bachelor's degree in Computer Science and Engineering (CSE) or equivalent from a reputed institution.


• Strong understanding of Large Language Models (LLMs), Embeddings, Vector Databases, Fine-tuning, Model Serving, TorchServe/Triton Inference Server/KServe (KFServing).


• Proficiency in programming languages such as Python.