SR. AI ML Engineer

Optimum Data Analytics; Pune City; On-site; INR 10,00,000 - 15,00,000; Full time

Job summary

A leading data analytics company in Pune is seeking an AI ML Engineer with 6-10 years of experience. The ideal candidate will have strong skills in building and deploying machine learning models, especially using tools like PySpark and Scikit-learn. Familiarity with Large Language Models (LLMs) such as OpenAI's GPT and cloud platforms is a plus. This is an onsite role requiring immediate joiners.

Qualifications

• 5-10 years of experience in AI/ML engineering.
• Proficiency in machine learning model development and evaluation.
• Hands-on experience with large-scale data processing using PySpark.

Responsibilities

• Design and deploy machine learning models for applications.
• Implement end-to-end ML pipelines using PySpark and MLflow.
• Collaborate with teams on ML model performance and improvements.

Skills

Python programming

Machine learning algorithms

Data processing with PySpark

MLflow

Pandas

NumPy

Scikit-learn

Large Language Models

Cloud platforms

Containerization technologies

Education

Bachelor's or Master's degree in Computer Science, AI/ML, Data Science

Tools

TensorFlow

PyTorch

AWS

Azure

Google Cloud

Docker

Kubernetes

Job description

Position: AI ML Engineer

Experience: 6-10 years

Location: Pune/Indore office

Work Mode: Onsite

Notice Period: Immediate Joiner 15 days

Job Summary:

We are looking for highly motivated and experienced AI ML Engineers to join our advanced analytics and AI team. The ideal candidates will have strong proficiency in building, training, and deploying machine learning models at scale using modern ML tools and frameworks. Experience with LLMs (Large Language Models) such as OpenAI and Hugging Face Transformers is highly desirable.

Key Responsibilities:

• Design, develop, and deploy machine learning models for real-world applications.
• Implement and optimize end-to-end ML pipelines using PySpark and MLflow.
• Work with structured and unstructured data using Pandas, NumPy, and other data processing libraries.
• Train and fine-tune models using scikit-learn, TensorFlow, or PyTorch.
• Integrate and experiment with Large Language Models (LLMs) such as OpenAI GPT, Hugging Face Transformers, etc.
• Collaborate with cross-functional teams including data engineers, product managers, and software developers.
• Monitor model performance and continuously improve model accuracy and reliability.
• Maintain proper versioning and reproducibility of ML experiments using MLflow.

Required Skills:

• Strong programming experience in Python.
• Solid understanding of machine learning algorithms, model development, and evaluation techniques.
• Experience with PySpark for large-scale data processing.
• Proficient with MLflow for tracking experiments and model lifecycle management.
• Hands-on experience with Pandas, NumPy, and Scikit-learn.
• Familiarity or hands-on experience with LLMs (e.g., OpenAI, Hugging Face Transformers).
• Understanding of MLOps principles and deployment best practices.
• Strong domain expertise in the AI area including LLM. Computer Vision, Time Series, RAG, fine-tune large models, traditional ML models, etc., (AI depth).
• Familiarity with cloud platforms (e.g., AWS, Azure, Google Cloud) and containerization technologies (e.g., Docker, Kubernetes) is a plus.

Preferred Qualifications:

• Bachelors or Masters degree in Computer Science, AI/ML, Data Science, or a related field.
• Experience in cloud ML platforms (AWS SageMaker, Azure ML, or GCP Vertex AI) is a plus.
• Strong analytical and problem-solving abilities.
• Excellent communication and teamwork skills.