JAKEX Tech

AI/ML Engineer

Pune, IndiaFull-time3-5 years₹15-28 LPA
PythonTensorFlowNLPComputer Vision

About the Role

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
  • 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

Interested?

Apply now and join our amazing team!

Not the Right Fit?

Browse all open positions or send us your resume for future opportunities.