AI/ML Engineer

We are seeking a highly skilled AI/ML Algorithm Engineer with specialized expertise in optimizing deep machine learning algorithms on NVIDIA GPUs. This role involves using advanced optimization techniques such as quantization and pruning to enhance the performance and efficiency of object detection algorithms.

Design, implement, and optimize deep learning algorithms for object detection, focusing on efficiency improvements through quantization, pruning, and other advanced techniques.

• Develop and deploy AI models on NVIDIA GPU platforms, ensuring optimal use of hardware resources.
• Develop the Camera and Lidar based object detection algorithms using CNNs and PointNet architectures.
• Optimize the Algorithms using different approaches like Quantization, Pruning etc.

Required Skills and Experience

Required Skills:

• Proven experience in optimizing deep learning algorithms using techniques like quantization and pruning.
• Proficiency in programming languages such as Python, C++, CUDA, and familiarity with deep learning frameworks like OpenCV, TensorFlow, PyTorch, or similar.
• Demonstrated experience with NVIDIA GPU programming and optimization, particularly on platforms like NVIDIA Jetson Xavier NX and NVIDIA Orin NX.
• Experience in converting ML models to efficient formats using ONNX or TensorRT for deployment on NVIDIA GPUs.
• Strong experience in image processing and deep learning, specifically with CNNs and PointNet architectures.

Desired Skills:

• Previous experience in a similar role within the tech industry, particularly in areas related to Autonomous Vehicles, Robotics, or IoT.