Return to jobs list

Head Of AI/Machine Learning

Job type: Full Time · Department: Tech · Work type: Hybrid

New Delhi, Delhi, India; Bengaluru, Karnataka, India

At Aftershoot, we’re building a global community of photographers who spend less time behind their screens and more time behind the lens. Every feature we create, every update we ship, is designed to make post-processing easier, faster, and smarter for photographers around the world.

From culling to editing to retouching.

Our AI tools do all the heavy lifting, so photographers can focus on what they do best: capturing moments, telling stories, and creating magic.

We’re a fast-growing, passionate team backed over 100K+ active users globally. Double digit ARR in less than 4 years, amazing investors and team who truly believe in our mission.

Our mission is clear: Become the default companion for every photographer in the world.

If you care deeply about building meaningful products, working with curious and driven teammates, and being part of a team that genuinely loves what they do, we’d love to meet you.

Let’s create the future of AI in photography, together.

🚀Your Mission

We are looking for someone to lead the most important function at Aftershoot, AI/Machine Learning. Your goal would be to create the vision & roadmap to ensure Aftershoot continues to lead with cutting edge technology. You will lead a team of researchers and engineers who are the backbone of our product & a blessing for our customers. You will work on frameworks & processes to improve image processing. You will work with the leadership to craft plans & goals for Aftershoot.

🎯What You Will Be Doing

  • Define the AI roadmap and architectural direction across all computer vision–driven product lines.

  • Lead end-to-end development of CV models—from research, dataset design, and training to evaluation and deployment.

  • Drive optimisation of deep learning models for edge environments through quantization, pruning, and distillation.

  • Build, mentor, and scale a high-performing ML research and engineering team.

  • Partner with Product and Engineering to translate AI advancements into user-facing features and workflows.

  • Establish technical standards, experimentation frameworks, and model quality benchmarks across the organisation.

💪What We’re Looking For

  • 10+ years of deep experience in Machine Learning, with a strong foundation in Computer Vision.

  • 2–4 years of hands-on experience in Image Processing, algorithms, and applied techniques.

  • Experience building and deploying AI systems for Edge / mobile / real-time use cases.

  • Strong background in designing scalable ML architectures, pipelines, and model optimisation workflows.

  • Proven history of leading teams, mentoring engineers, and setting technical direction.

  • Prior experience in startups (fast, iterative environments where you ship fast and often).

Bonus Points:

  • Experienced with 3D computer vision, video processing, and text-to-video, image-to-video generation.

  • Experienced working with highly skewed and imbalanced data.

  • A genuine passion for photography or visual creativity (huge bonus).

What happens after you apply?

  1. Step 1: If your profile aligns with our vibe and requirements for the role, our people team will reach out to you for a brief phone call.

  2. Step 2: If everything aligns, your next chat will be with one of your future team members to better understand the role and your experience.

  3. Step 3: You will then be given a task - this helps us evaluate your thought process and help you showcase your expertise.

  4. Step 4: If everything goes well, you will meet the leadership for a quick bar-raiser round.

  5. Step 5: If you do manage to clear all rounds, this is the point we make an offer, shake hands, and welcome you to the team.

You will work with:

  1. Team : Nikhil, Sumandeep, Karan, Jaidev, Himanshu

Made with