Autonomous Driving algorithm engineer | AI for Edge devices (SoC)
AI + SoC
Optimizing computer vision models for the edge.
- Designed End-to-End solution for Bird's eye view (BEV) Perception algorithm both training and deployment on Nvidia Orin AGX.
- Selection of algorithm and trained BEV segmentation(free space and boundary) on custom dataset of mines.
- Architected everything for inference on Nvidia Drive Orin SoC.
- Optimization of overall pipeline.
- Inference hardware and camera selection.
- Camera Calibration
- Data Collection strategy support.
Pytorch, C++, TensorRT C++, OpenCV cuda, cuda kernel, Driveworks, DriveOS, Computer vision
e.g. Implemented highly parallelized computer vision function from scratch on Vector processor(DSP) like WarpAffine, crop_resize, nms, cosine_d, cv pre/post-processing algorithms(yolo v5/v7/x, centernet, reid, ggcnn, segmentation, inpainting etc). Achieved highest inference runtime for the aforementioned functions.
C/C
Python
OpenCL
OpenCV
DSP
Computer Vision
Nvidia Drive OS
Nvidia DriveWorks