Computer Vision & AI Intern

Company Description

InfiVR specializes in developing cutting-edge Enterprise solutions powered by Virtual Reality (VR), Augmented Reality (AR), Artificial Intelligence (AI), Digital Twin, and Metaverse technologies. With expertise in industries such as Aviation, Aerospace, Automobiles, Defence, Pharmaceuticals, and Healthcare, InfiVR creates innovative solutions for virtual training, simulations, and collaborative design. Serving Fortune 500 clients globally, including leading Defence organizations, InfiVR ensures end-to-end support from conceptualization to implementation. Our mission is to empower the future workforce with advanced technologies, improving productivity, accuracy, and safety across industries. Through a robust partner network, we deliver solutions across numerous countries, including the US, Germany, Singapore, and UAE.

We are looking for a highly motivated Computer Vision & AI Intern to work on cutting-edge ML, Deep Learning, and Vision Intelligence problems.

You will work closely with senior engineers and researchers on real-world applications such as vision-based detection, tracking, action recognition, XR-AI integration, and intelligent automation.

This is a hands-on engineering role, not a theoretical internship.

• Design, implement, and optimize computer vision pipelines using OpenCV and Python
• Train and fine-tune deep learning models for tasks such as:
• 
  
• Object detection & segmentation
• Pose / action recognition
• Face & gesture analysis
• Image classification & tracking
• Work with modern ML architectures, including:
• 
  
• CNNs (ResNet, EfficientNet, MobileNet)
• Transformers / Vision Transformers (ViT, DETR, SAM, CLIP, etc.)
• Prepare datasets, perform data annotation, augmentation, and preprocessing
• Train models using PyTorch / TensorFlow
• Evaluate models using proper metrics and validation techniques
• Optimize models for real-time inference (edge / mobile / XR devices)
• Collaborate with AR/VR and backend teams to integrate AI models into applications
• Document experiments, results, and learnings clearly

• Strong proficiency in Python
• Hands-on experience with OpenCV
• Solid understanding of Computer Vision fundamentals
• Experience with Deep Learning frameworks:
• 
  
• PyTorch (preferred) or TensorFlow
• Familiarity with:
• 
  
• CNNs, transfer learning
• Model training & fine-tuning
• Understanding of ML pipelines (data training evaluation inference)

• Experience with Transformers / Vision Transformers
• Familiarity with Hugging Face ecosystem
• Exposure to YOLO, MediaPipe, SAM, Detectron2
• Experience with video analytics
• Knowledge of model optimization (ONNX, TensorRT, quantization)
• Basic understanding of AR/VR, Unity, or real-time systems
• Experience working with Kaggle or research projects

Students or fresh graduates in:

• Computer Science
• AI / ML
• Data Science
• Robotics
• Electronics / Mechatronics (with strong CV skills)

Candidates who:

• Enjoy building & experimenting
• Can work independently and take ownership
• Are curious about real-world AI problems, not just academic ones

6 Months (Extendable based on performance)