AI/ML Engineer

Jakex Group; Maharashtra; On-site; INR 1,500,000 - 2,800,000; Full time

Benefits offered by this job

Competitive salary with equity options

Health insurance for you and your ...

Job summary

JAKEX Tech in Pune, India is seeking a Machine Learning Engineer to design production models for NLP, computer vision, and analytics across education, marketing, and enterprise software.

You will build ML pipelines, collaborate with product teams, and apply the latest AI research to deliver scalable solutions. 3-5 years of experience, Python, and TensorFlow/PyTorch are expected.

Qualifications

• 3-5 years of experience in machine learning engineering.
• Strong proficiency in Python and ML frameworks (TensorFlow, PyTorch).
• Experience with NLP techniques and libraries (spaCy, Hugging Face).
• Knowledge of computer vision and image processing.
• Experience with cloud ML services (AWS SageMaker, GCP AI Platform).
• Understanding of MLOps practices.
• Strong mathematical foundation (linear algebra, statistics).
• Excellent problem-solving and analytical skills.

Responsibilities

• Design and implement machine learning models for production use.
• Work on NLP solutions for our EduTech platform.
• Develop computer vision applications for various clients.
• Build and maintain ML pipelines and infrastructure.
• Collaborate with product teams to integrate AI features.
• Optimize models for performance and scalability.
• Stay current with latest AI/ML research and techniques.
• Document models and create technical specifications.

Skills

Python

TensorFlow

PyTorch

spaCy

Hugging Face

NLP

Computer Vision

Cloud ML

MLOps

Mathematics

Tools

AWS SageMaker

GCP AI Platform

spaCy

Hugging Face

Job description

About The Role

Join our AI team to build intelligent solutions that power our products across education, marketing, and enterprise software. You're work on exciting projects involving NLP, computer vision, and predictive analytics.

What You'll Do

• Design and implement machine learning models for production use
• Work on NLP solutions for our EduTech platform
• Develop computer vision applications for various clients
• Build and maintain ML pipelines and infrastructure
• Collaborate with product teams to integrate AI features
• Optimize models for performance and scalability
• Stay current with latest AI/ML research and techniques
• Document models and create technical specifications

What We're Looking For

• 3-5 years of experience in machine learning engineering
• Strong proficiency in Python and ML frameworks (TensorFlow, PyTorch)
• Experience with NLP techniques and libraries (spaCy, Hugging Face)
• Knowledge of computer vision and image processing
• Experience with cloud ML services (AWS SageMaker, GCP AI Platform)
• Understanding of MLOps practices
• Strong mathematical foundation (linear algebra, statistics)
• Excellent problem-solving and analytical skills

Nice to Have

• +Experience with LLMs and prompt engineering
• +Published papers or research in AI/ML
• +Experience with real-time ML systems
• +Knowledge of edge deployment for ML models

Benefits

• Competitive salary with equity options
• Health insurance for you and your family
• Flexible work arrangements
• Conference sponsorships and research time
• Access to GPU clusters for experiments
• Collaborative and innovative work environment

Quick Info

Department: JAKEX Tech

Location: Pune, India

Employment Type: Full-time

Experience: 3-5 years

Salary Range: ₹15-28 LPA