Senior Machine Learning Engineer

Aftershoot; New Delhi; On-site; INR 3,000,000 - 6,000,000; Full time

Job summary

Aftershoot in New Delhi, India seeks an experienced Machine Learning Engineer specializing in Computer Vision and Generative AI to own end-to-end development of innovative vision problems. You will lead projects from ideation through deployment, ensuring robust ML systems and efficient monitoring post-launch.

You will collaborate with product, backend, and platform teams to define timelines and drive execution while continuously improving model performance and cost efficiency.

Qualifications

• 5–8 years of experience in Deep Learning, Computer Vision and Generative AI.
• Hands-on with CNNs, GANs, VAEs, diffusion, and text-to-image generation.
• Proficient in Python and ML frameworks (TensorFlow, PyTorch, JAX).
• Experience deploying Vision models on edge devices and optimizing for resource constraints.
• Strong problem-solving and ability to work with minimal supervision.

Responsibilities

• Own the lifecycle of Vision and GenAI problems from concept to deployment.
• Design, develop, and deploy scalable ML systems for Vision use-cases.
• Optimize training and inference pipelines to maximize GPU utilization and minimize costs.
• Collaborate with Product, Backend, and Platform teams to align timelines with business goals.

Skills

Deep Learning

Computer Vision

Generative AI

CNNs

GANs

VAEs

Diffusion models

Inpainting

Python

TensorFlow

PyTorch

JAX

Edge deployment

GPU optimization

Education

B.Tech/MS in CS/ML

Tools

TensorFlow

PyTorch

JAX

Docker

Kubernetes

Job description

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 an experienced Machine Learning Engineer with deep expertise in Computer Vision and Generative AI to join our team in New Delhi, India, and own the end-to-end development and improvement of various innovative problem statements. You will take full ownership of projects, from ideation to delivery, ensuring successful deployment as well as ML monitoring post-launch.

What You Will Be Doing

• Own the entire lifecycle of the Vision and GenAI problem statement, from initial concept throughout deployment.
• Research, design, develop, and deploy robust and scalable ML systems for various Vision use-cases.
• Optimize model training and inference pipelines to maximize GPU utilization and minimize costs.
• Collaborate with Product, Backend, and Platform teams to define project timelines, ensure alignment of business goals, and drive strong execution.

What We’re Looking For

• Solid foundation in Deep Learning, Computer Vision, and Generative AI, with a proven experience of 5-8 years, preferably in a fast-paced startup environment.
• In-depth and practical knowledge of CNNs, GANs, VAEs, Diffusion models, and Inpainting methods, image processing techniques, text-to-image and image-to-image generation architectures, etc.
• Strong programming skills in Python and proficiency with ML frameworks (e., Tensorflow, Pytorch, JAX) - create code that is understandable, simple, clean, and easily shared with others.
• Experience in deploying Vision models on edge devices, optimizing for resource constraints.
• Knack for staying up to date with the latest research and trying out unconventional, out-of-the-box ideas.
• Passion for problem-solving and creative thinking, with the ability to break down complex and abstract problems into actionable items.
• Self-motivated to build, with an ability to thrive with minimal oversight and process.

Bonus Points

• Experienced with 3D computer vision, video processing, and text-to-video, image-to-video generation.
• Knowledge of the Rust programming language for implementing the inference pipelines.
• Experienced working with highly skewed and imbalanced data.
• Familiarity with cloud platforms (AWS, GCP, Azure) and containerization (Docker, Kubernetes).

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. Captain : Hardik
2. Team: Pratyush, Abinash, Abhishek, Karan