Vision Deep Learning Engineer - Pytorch

Get AI-powered advice on this job and more exclusive features.

As a Vision Machine Learning / Software Engineer, you will play a key role in enhancing Mentium's deep learning library for vision AI models on our in-memory compute ultra-low power hardware. Collaborating closely with a small team of software and hardware experts, you will design and refine a client-facing codebase that enables seamless compilation and deployment of vision models for real-time applications. Additionally, you will support the training and testing of models for demonstrations.

A strong understanding of vision model components is essential, including convolutional layers, transformers, activation functions, pooling, normalization, and non-maximum suppression. Familiarity with the structure, training and quantization of models such as YOLO, Mask R-CNN, MobileNet, ConvNeXt, EfficientNet, and Vision Transformers is required. You will work with libraries like Torchvision, Ultralytics, MMCV, Keras, ONNX, and OpenCV, adapting to client needs. Awareness of hardware fundamentals—such as memory transfer, inference time, model size, and precision—is critical for real-time edge deployments.

Proficiency in Python is essential, including expertise in OOP principles, decorators, context managers, and numerical computing libraries like NumPy and pandas. You should demonstrate strong skills in modular development, collaborative coding using GitHub, and producing clear, well-structured documentation to enhance the customer experience.

This role offers an exciting opportunity to work with cutting-edge vision AI technology and make a meaningful impact on the development of a robust, client-facing software library.

Requirements and Qualifications :

• Educational Background : Master's or PhD degree in Computer Science, Electrical Engineering, or a related field.
• Programming Expertise : Proficiency in Python and PyTorch, including object-oriented programming principles, decorators, context managers, and experience with numerical computing libraries such as NumPy and pandas.
• Vision Model Proficiency : Strong understanding of components like convolutional layers, transformers, activation functions, pooling, normalization, and non-maximum suppression.
• Model Experience : Familiarity with the structure, training, and quantization of models such as YOLO, Mask R-CNN, MobileNet, ConvNeXt, EfficientNet, and Vision Transformers.
• Frameworks and Libraries : Experience with libraries like Torchvision, Ultralytics, MMCV, Keras, ONNX, and OpenCV.
• Hardware Awareness : Understanding of hardware fundamentals, including memory transfer, inference time, model size, and precision, crucial for real-time edge deployments.
• Development Practices : Proficiency in modular development, collaborative coding using GitHub, and producing clear, well-structured documentation.
• Soft Skills : Strong problem-solving and analytical skills, excellent communication abilities, and a collaborative mindset to work effectively within a multidisciplinary team.
• Familiarity with OpenCV programming and video processing pipelines is a plus.
• Having a knowledge and experience on working with Camera drivers and camera pipelines using hardware accelerators such as ISPs is also a plus.

Why Join Mentium?

At Mentium, you'll be at the forefront of vision AI technology, working alongside a talented team dedicated to innovation. We offer a collaborative environment where your contributions directly influence the development of cutting-edge solutions. Join us to make a meaningful impact in the field of vision AI.

Seniority level

Seniority level

Entry level

Employment type

Employment type

Full-time

Job function

Job function

Engineering and Information Technology

Get notified about new Deep Learning Engineer jobs in California, United States

Mountain View, CA $206,300.00-$305,600.00 1 week ago

Senior Software Engineer, PyTorch - Deep Learning

Senior Deep Learning Software Engineer, LLM Performance

Principal Deep Learning Software Engineer, LLM Performance

San Francisco, CA $177,000.00-$245,000.00 1 month ago

California, United States $224,000.00-$356,500.00 2 days ago

San Francisco Bay Area $60.00-$120.00 3 weeks ago

Senior Deep Learning Compiler Engineer - XLA

California, United States $148,000.00-$287,500.00 15 hours ago

Senior Software Engineer, TensorRT Inference

California, United States $184,000.00-$356,500.00 1 week ago

Threat Detection Research Engineer (Remote)

Sunnyvale, CA $110,000.00-$180,000.00 2 weeks ago

Senior Software Engineer, Streaming Protocols

California, United States $184,000.00-$356,500.00 1 week ago

Senior Software Engineer, Registry - US (Remote)

San Francisco, CA $177,000.00-$245,000.00 2 months ago

Senior Software Engineer, Workspaces - US (Remote)

San Francisco, CA $177,000.00-$245,000.00 2 months ago

California, United States $148,000.00-$287,500.00 1 day ago

Senior Software Engineer, Networking Software

California, United States $148,000.00-$287,500.00 1 week ago

Software Engineer, Billing & Expansion Team - US (Remote)

San Francisco, CA $146,000.00-$203,000.00 1 month ago

Staff Software Engineer, Metrics - US (Remote)

San Francisco, CA $210,000.00-$290,000.00 2 months ago

We're unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.

J-18808-Ljbffr