Experience Required: 3–5 Years
Job Type: Full-time (Night Shift)
Department: Engineering / Artificial Intelligence
Work Location: Remote
Salary: 18 LPA to 23 LPA

Job Summary

We are looking for an experienced Machine Learning Engineer with 3–5 years of hands-on industry experience to work on advanced AI/ML systems, model optimisation, and machine learning infrastructure.

The ideal candidate should have a strong engineering background and the ability to understand, analyse, and improve how machine learning models perform in real-world environments. The role requires practical knowledge of model inference, performance optimisation, GPU execution, and production ML systems.

The candidate will also contribute to AI/ML training and evaluation initiatives by assessing technical solutions, identifying model limitations, validating outputs, and providing structured feedback to improve AI systems.

Key Responsibilities

- Design, develop, test, and optimise machine learning models and AI/ML systems.
- Work on model inference, model serving, performance optimisation, and ML infrastructure.
- Analyse GPU execution, CUDA operations, kernel optimisation, and computational efficiency.
- Evaluate model performance based on speed, memory consumption, compute requirements, accuracy, and numerical stability.
- Identify performance bottlenecks and recommend improvements in ML pipelines and model execution.
- Evaluate AI-generated code, technical solutions, and machine learning outputs for accuracy and reliability.
- Develop and execute test cases to assess the performance, consistency, and efficiency of AI/ML systems.




- Identify technical errors, inconsistencies, edge cases, and limitations in machine learning implementations.
- Contribute to AI/ML training initiatives by evaluating model responses, preparing technical problem statements, and providing structured feedback.
- Research emerging machine learning technologies, frameworks, and optimisation techniques.
- Document technical findings, optimisation strategies, and implementation approaches.
- Collaborate with engineering teams to improve the performance and reliability of AI systems.

Required Skills & Qualifications

- Bachelor's or Master's degree in Computer Science, Artificial Intelligence, Machine Learning, Data Science, or a related technical field.
- 3–5 years of relevant professional experience in Machine Learning, Deep Learning, AI Engineering, or ML Infrastructure.
- Strong programming skills in Python.
- Hands-on experience with machine learning frameworks such as PyTorch or TensorFlow.
- Strong understanding of machine learning algorithms, model development, inference, and deployment.
- Practical experience with model optimisation, performance benchmarking, or production ML systems.
- Understanding of GPU computing, CUDA, kernel optimisation,



or inference engines.
- Strong analytical, debugging, and problem-solving abilities.
- Ability to evaluate technical trade-offs involving speed, memory, compute, accuracy, and numerical stability.
- Excellent research, technical communication, and documentation skills.

Preferred Skills

- Experience working with Large Language Models (LLMs) and Generative AI.
- Exposure to GPU architecture, distributed computing, or high-performance computing.
- Experience with inference optimisation tools and frameworks such as ONNX, TensorRT, or vLLM.
- Familiarity with model benchmarking, evaluation, and AI system performance testing.
- Experience in AI model training, technical evaluation, or AI-assisted development.

Ideal Candidate Profile

We are looking for someone who:

- Has a strong engineering mindset and enjoys solving complex technical challenges.
- Can independently research, analyse, and validate technical implementations.
- Questions assumptions and investigates technical details rather than accepting solutions at face value.
- Understands the practical behaviour of ML systems beyond theoretical concepts.
- Is comfortable working with advanced AI technologies and continuously learning new tools and techniques.

Pay: ₹1,800,000.00 - ₹2,300,000.00 per year

Advantages:

- Work from home

Education:

- Bachelor's (Required)

Experience:

- Machine learning: 3 years (Preferred)

Language:

- English (Required)

Shift availability:

- Night Shift (Required)

Work Location: Remote

📌 Machine Learning Engineer – AI/ML Systems & Model Optimisation (India)
🏢 E-Commerce Collections
📍 India