Machine Learning Intern
Full Time · Tech · On-Site
New Delhi, Delhi, India
At Aftershoot, we’re building more than just AI tools; 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 worldwide, 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 by over 30,000+ photographers and $10 M+ in ARR.
But we’re just getting started. Our mission is clear: solve the biggest pain points in photography, and keep evolving with photographers who use our tools daily.
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.
We're looking for a passionate and driven Machine Learning Intern to join our team in New Delhi for 6 months.
🚀 Your Mission
You'll work closely with our experienced ML engineers to research, prototype, and build innovative
solutions in Computer Vision and Generative AI. This is a hands-on opportunity to contribute to
real-world projects that push the boundaries of visual understanding and generation.
🎯 What You Will Be Doing
Assist in the design, development, and evaluation of deep learning models for vision-based use cases.
Experiment with architectures like CNNs, GANs, VAEs, and diffusion models.
Contribute to data preprocessing, augmentation, and visualization workflows.
Collaborate with cross-functional teams (Product, Engineering, Design) to iterate quickly on
ideas.
Conduct literature reviews and apply the latest research techniques in Generative AI and
image processing.
Help build reproducible training/inference pipelines using PyTorch or TensorFlow.
Participate in code reviews, team discussions, and knowledge-sharing sessions.
💪 What We’re Looking For
Currently pursuing or recently completed a Bachelor’s/Master’s degree in Computer
Science, Electrical Engineering, AI, or related field.
Strong understanding of Deep Learning fundamentals, especially CNNs and image-based
architectures.
Familiarity with one or more of the following: GANs, VAEs, diffusion models, image
segmentation, inpainting, text-to-image models.
Proficient in Python, with experience in PyTorch or TensorFlow.
Hands-on experience with at least one computer vision or generative model project (coursework or personal).
Ability to write clean, maintainable, and modular code.
Eagerness to learn, experiment, and build in a fast-paced environment.
⭐ Bonus Skills
Experience with OpenCV, Hugging Face, or ONNX.
Exposure to tools like Weights & Biases, Streamlit, or FastAPI.
Contributions to GitHub repositories or Kaggle competitions.
Familiarity with optimization techniques for model inference on edge devices.
What You'll Gain
Mentorship from experienced ML engineers working on state-of-the-art problems.
Exposure to the full ML lifecycle – from research to deployment.
Real-world experience with impactful, production-grade vision systems.
Opportunity to convert into a full-time role based on performance in these 6 months.
Apply With
Resume.
GitHub/Portfolio (if available).
Brief paragraph on why you’re interested in this role.
Autofill application
Save time by importing your resume in one of the following formats: .pdf or .docx.