Company: Agrex.ai | Delhi, India

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

The post Agrex.ai hiring Machine Learning Engineer | Apply now appeared first on Job4Everyone.