Computer Vision AI & ML Engineer

2 months ago

Bengaluru, IndiaMid Level

H1B Sponsor

Responsibilities

Develop and optimize deep learning models for object detection, segmentation, tracking, and 3D scene understanding.
Build scalable data processing, training, evaluation, and deployment pipelines for real-time systems.
Design annotation strategies and tooling for automated labeling, QA workflows, dataset management, augmentation, and versioning.
Implement uncertainty estimation, failure detection, monitoring, reliability, and automated performance reporting frameworks.
Conduct proof-of-concept experiments and translate research findings into practical perception prototypes.
Collaborate with robotics, systems, and simulation teams to integrate perception models into production pipelines and improve end-to-end performance.

Strong experience with PyTorch, TensorFlow, or JAX.
Background in computer vision tasks including detection, segmentation, tracking, or 3D scene understanding.
Proficiency in Python; familiarity with C++ is a plus.
Experience building training pipelines, evaluation frameworks, and ML deployment workflows.
Knowledge of 3D geometry, sensor processing, or multimodal sensor fusion involving RGB-D, LiDAR, or stereo data.
Experience with data annotation tools, dataset management, and augmentation techniques.
Familiarity with robotics, simulation environments, or real-time systems.
Understanding of uncertainty modeling, reliability engineering, or ML monitoring and MLOps practices.

11-50 employees

Building general purpose robotic intelligence.