We are looking for a hands-on Machine Learning Engineer to join our core technology team and work on an early-stage Computer Vision & Generative AI product. The role involves building ML pipelines, preparing and managing datasets, training and fine-tuning models, and developing reliable computer-vision solutions for real-world applications. This is an ownership-driven role where you will work closely with the founding and engineering team.
Key Responsibilities
- Build and improve Computer Vision and Deep Learning pipelines
- Prepare, clean, structure and annotate ML datasets
- Develop image-processing and computer-vision solutions
- Train, fine-tune and evaluate deep-learning models
- Work with Generative AI and image-generation models
- Experiment with different model architectures and approaches
- Optimize models for efficient GPU inference
- Build and evaluate working ML prototypes
- Integrate trained models into the existing application stack
- Monitor model performance and continuously improve results
Required Skills
- Strong Python programming
- Strong hands-on experience with PyTorch
- Solid Machine Learning & Deep Learning fundamentals
- Strong Computer Vision & OpenCV experience
- Experience with image processing, segmentation and object detection
- Experience with human pose estimation / human parsing
- Experience with ML dataset preparation, cleaning and annotation
- Experience with model training, fine-tuning and evaluation
- Experience with Generative AI / Diffusion Models
- Practical experience with GPU / CUDA
Valuable to Have
- Image-to-image generation
- Video processing / generation
We are looking for someone who can build and experiment, not just work with existing APIs or tutorials. You should be comfortable taking an ML problem from: Dataset → Training → Testing → Evaluation → Optimization → Deployment. Practical experience and demonstrable projects will be valued more than certifications.
Who Should Apply?
You are a strong fit if you:
- Have built real-world ML / Computer Vision projects
- Have hands-on experience working with datasets
- Have trained or fine-tuned ML models yourself
- Can demonstrate your work through GitHub, portfolio or previous projects
- Are comfortable working independently
- Enjoy solving technically challenging problems in an early-stage startup environment
Qualifications
- 2–4 years of relevant professional or strong project experience
- B.Tech / M.Tech / MS in Computer Science, AI, ML or a related field preferred
- Strong analytical and problem-solving skills
Why Join Us?
- Work on an early-stage AI technology product
- High technical ownership
- Opportunity to build ML systems from the ground up
- Direct collaboration with the founding team
- Research and experimentation-driven environment
- Opportunity to take models from prototype to real-world deployment