AI/ML Engineer

JAKEX Group; Pune District; On-site; INR 1,200,000 - 1,800,000; Full time

Benefits offered by this job

Equity options

Health insurance

Conference sponsorships

GPU access

Job summary

JAKEX Group in Pune, India is seeking a Machine Learning Engineer to design and productionize ML models for our education, marketing, and enterprise software products. You’ll work on NLP, computer vision, and predictive analytics.

You will collaborate with product teams, optimize models for performance, and build ML pipelines with MLOps best practices. Strong Python skills, TensorFlow or PyTorch, and experience with NLP and CV are required; equity options and health insurance are offered.

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.

Responsibilities

• Design and implement machine learning models for production use.
• Work on NLP solutions for EduTech platform.
• Develop computer vision applications for 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 AI/ML research and techniques.
• Document models and create technical specifications.

Skills

Python

ML frameworks

NLP

Computer vision

MLOps

Problem solving

Tools

TensorFlow

PyTorch

SpaCy

Hugging Face

AWS SageMaker

GCP AI Platform

Job description

Join our AI team to build intelligent solutions that power our products across education, marketing, and enterprise software. You'll 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
• 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
• Competitive salary with equity options
• Health insurance for you and your family
• Conference sponsorships and research time
• Access to GPU clusters for experiments
• Collaborative and innovative work environment