Senior ML Engineer - Computer Vision

Location: New Delhi, India / Remote

About Us:
Aftershoot is a cutting-edge AI and ML-powered photography tool designed to streamline the post-shoot flow. We are dedicated to empowering photographers worldwide by simplifying their workflow and enhancing their creative processes. Our goal is to make post-shoot tasks a breeze, enabling photographers to focus on what they do best: capturing captivating moments through their lenses, letting Aftershoot take care of the rest!

About the Role:
We are looking for a Senior Machine Learning Engineer with deep expertise in Computer Vision and Generative AI to join our team and lead the end-to-end development and improvement of various innovative products. You will take full ownership of projects, from ideation to delivery, managing resources and timelines, ensuring successful deployment as well as ML monitoring post launch.

Responsibilities:

• Lead the entire lifecycle of Vision and GenAI projects, 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 roadmaps, ensure alignment of business goals, and drive strong execution
• Mentor and guide junior and intermediate engineers, fostering a culture of innovation and excellence

About You:

• Solid foundation in Deep Learning, Computer Vision and Generative AI, with a proven experience of 6-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.g 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
• Demonstrated experience in leading projects and teams, with strong communication skills

Bonus Points:

• Experienced with 3D computer vision, video processing and text-to-video, image-to-video generation
• Knowledge of 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)

Why Join Us?

• Make a real impact on how millions of photographers deliver their photos
• Work with a fun, talented team (we're pretty awesome, if we do say so ourselves)
• Competitive salary and benefits to keep you happy and healthy
• Flexible work hours and remote work options

We're looking for people with a can-do attitude and a mindset to do something that impacts an industry as a whole! If you don't want to be yet another cog in the system and want to work on innovative and challenging problems, we'll be excited to have you! 🚀