Machine Learning Engineer In-Office Internship
Company: Agrex.ai | Delhi, India
| Stipend | ₹ 8K – 15K |
| Duration | 3 Months |
| Location | Delhi, India |
| Job Offer | ₹ 4 LPA – 6.5 LPA |
| Mandatory Skills | Python, PyTorch, NumPy, TensorFlow, CUDA |
About Agrex.ai
Agrex.ai is a video analytics company transforming existing CCTV infrastructure into intelligent, real-time monitoring systems for retail, manufacturing, banking, logistics, and education. Our platform delivers people & vehicle analytics, SOP/compliance monitoring, and operational insights—on edge devices (e.g., Jetson) and in the cloud.
Role Overview
We’re looking for a Computer Vision Intern who’s excited to learn and contribute to real-world video analytics systems. You’ll get hands-on experience with training and optimizing models, building video pipelines, and deploying on Jetson and NVIDIA GPUs—while working closely with our Product and Engineering teams.
Responsibilities
- Assist in implementing, training, and evaluating detection/segmentation/tracking models (YOLO, RTDETR, Mask models, ByteTrack, OC-SORT).
- Help integrate models into NVIDIA DeepStream / GStreamer pipelines; export to ONNX/TensorRT and optimize for speed & efficiency.
- Contribute to features like people/vehicle counting, queue analytics, intrusion detection, PPE checks, and re-ID.
- Work with real video streams (RTSP/H.264/H.265).
- Support dataset preparation: curation, labeling, augmentation.
- Track experiments with metrics (mAP, recall, FPS, GPU utilization).
- Debug and test field issues with the team.
Requirements
- Good coding skills in Python (NumPy, OpenCV, PyTorch/TensorFlow).
- Understanding of Computer Vision/ML concepts (convolutions, IoU, NMS, augmentation).
- Exposure to object detection, segmentation, or tracking.
- Comfortable learning from documentation and research papers.
Bonus Points
- Familiarity with NVIDIA stack: CUDA, TensorRT, DeepStream.
- Hands-on with Jetson or other edge devices.
- Knowledge of GStreamer/FFmpeg and video codecs.
- Experience with DVC, W&B, Docker, or ML pipelines.
- Interest in re-ID, pose estimation, OCR, or multi-camera systems.
Education
B.Tech/BE/M.Tech in CS/EE/Maths or equivalent practical experience
Extra Benefits
- Letter of Recommendation
- Certificate
- Flexible Hours
- Job Offer
