[Remote] [Remote in US] AI Deep Learning/ML Engineer

Note: The job is a remote job and is open to candidates in USA. Mentium Technologies Inc. develops vision AI technology for ultra-low-power in-memory computing hardware. The Vision Machine Learning / Software Engineer will enhance a client-facing deep learning library, compile and deploy vision models for real-time edge applications, and support model training, testing, and demonstrations.

Responsibilities

• 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

Skills

• • 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

Benefits

• Competitive compensation packages
• Opportunity to work on diverse, cutting-edge AI projects across a range of industries.
• 401(k)
• Flexible PTO
• Full PPO medical, dental, and vision insurance coverage

Company Overview

We are developing the next paradigm in AI computation, to help our most important muscle, the brain, to go further and open new unthinkable landscapes, new opportunities and new jobs. It was founded in 2017, and is headquartered in Santa Barbara, California, USA, with a workforce of 2-10 employees. Its website is https://mentium.tech.

Company H1B Sponsorship

Mentium Technologies Inc. has a track record of offering H1B sponsorships, with 1 in 2026, 2 in 2023. Please note that this does not guarantee sponsorship for this specific role.