Senior Machine Learning Engineer

Aftershoot; Delhi; On-site; INR 1,500,000 - 2,500,000; Full time

Job summary

Aftershoot is seeking an experienced Machine Learning Engineer in New Delhi with expertise in Computer Vision and Generative AI. This role involves owning the entire lifecycle of innovative ML projects, from ideation to delivery. Candidates should have 5-8 years of experience, strong programming skills in Python, and practical knowledge of deep learning architectures. Additional experience in deploying Vision models and optimizing workflows is preferred. Join a dynamic team focused on developing cutting-edge AI solutions.

Qualifications

• 5-8 years of proven experience in Deep Learning, Computer Vision, and Generative AI.
• Practical knowledge of CNNs, GANs, VAEs, and image processing techniques.
• Experience deploying Vision models on edge devices.

Responsibilities

• Own the lifecycle of the Vision and GenAI problem statement from concept to deployment.
• Research and develop robust ML systems for Vision use-cases.
• Collaborate with teams to define project timelines and align business goals.

Skills

Deep Learning

Computer Vision

Generative AI

Python

TensorFlow

PyTorch

JAX

Job description

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 5‑8 years of proven experience, preferably in a fast‑paced startup environment.
• In‑depth and practical knowledge of CNNs, GANs, VAEs, Diffusion models, Inpainting methods, image processing techniques, and text‑to‑image or image‑to‑image generation architectures.
• Strong programming skills in Python and proficiency with ML frameworks such as TensorFlow, PyTorch, and JAX; code should be understandable, simple, clean, and easily shared.
• Experience deploying Vision models on edge devices and optimizing for resource constraints.
• Knack for staying up‑to‑date with the latest research and experimenting with unconventional ideas.
• Passion for problem‑solving and creative thinking, with the ability to break down complex problems into actionable items.
• Self‑motivated and able to thrive with minimal oversight.

Bonus Points

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