Role Overview
We are looking for an ML / AI Intern interested in building practical AI systems. You will work with our engineering team on real-world projects involving areas such as computer vision, machine learning, language models, data processing, and deployment. We are looking for someone with good fundamentals, curiosity, and the ability to learn and build.
Responsibilities
- Model Development: Design, train and fine-tune deep learning models for images, text and audio using frameworks like PyTorch and TensorFlow.
- Computer Vision: Work on image processing, classification, object detection, tracking, or similar vision tasks depending on project requirements.
- Data Pipelines: Curate, clean, annotate and preprocess multimodal datasets to ensure high-quality inputs for training and evaluation.
- Model Optimization: Compress and optimize models for edge deployment using quantization, pruning and conversion to TensorRT / TFLite / ONNX.
- Model Integration: Develop Python scripts to integrate models with real-time video streams, text APIs, voice inferences and other sensor data where relevant.
- Benchmarking: Measure FPS, accuracy, latency and thermal behavior on target hardware and iterate on models to improve efficiency.
- Documentation: Record experiments, code, and results in a clear, concise format and stay updated with relevant AI/ML techniques.
Requirements
- Python: Comfortable writing Python code and familiar with basic data structures, functions, modules, and debugging.
- ML Fundamentals: Basic understanding of machine learning concepts such as training, validation, overfitting, evaluation metrics, and common model architectures.
- Deep Learning Frameworks: Some hands-on exposure to PyTorch, TensorFlow, or similar frameworks through coursework, projects, or self-learning.
- Computer Vision: Basic familiarity with image processing or computer vision concepts using OpenCV or similar tools.
- Development Tools: Basic familiarity with Git/GitHub and willingness to work with Linux and command-line tools.
- Learning Ability: Comfortable learning unfamiliar tools, libraries, and concepts while working on practical engineering problems.
Skills
- Python
- PyTorch
- TensorFlow
- OpenCV
- Git
Nice to Have
- Computer Vision Models: Exposure to YOLO, ResNet, MobileNet, object detection, tracking, pose estimation, or similar models.
- NLP / LLMs: Experience with embeddings, transformers, semantic search, text classification, information extraction, or chatbots.
- Edge Hardware: Projects involving Raspberry Pi, NVIDIA Jetson, etc.
- Deployment: Familiarity with ONNX Runtime, TensorRT, TensorFlow Lite, etc.
- DevOps: Basic understanding of Docker or containerized applications.
Benefits
- Real-World AI Experience
- End-to-End Exposure
- Breadth in AI
- Production Mindset
- Industry Exposure