Embedded Machine Learning Engineer Intern/Co-Op

Job Description

About Invisible AI

Invisible AI is a company building the future of computer vision by developing an end-to-end platform that digitizes manufacturing operations. They deploy edge AI cameras to automate and improve the accuracy, reliability, and safety of manual assembly work in factories. The founders, with a background in self-driving car technology, bring years of experience in large-scale AI & Machine Learning deployment. Invisible AI aims to deliver the power of computer vision to real-world customers.

Embedded Machine Learning Engineer Intern/Co-Op Job Description

As an Embedded Machine Learning Engineer Intern/Co-Op at Invisible AI, you will work with cutting-edge technologies to:

• Validate the performance of the company’s machine learning stack on various off-the-shelf hardware accelerators and deep learning inference platforms.
• Contribute to the development of the next generation of hardware by tackling challenges related to deploying machine learning models built in different libraries on edge compute platforms.
• Analyze the feasibility, computational and runtime performance of machine learning models on these platforms.
• Work alongside a world-class team of engineers to deploy a new wave of AI products that work seamlessly across domains without requiring weeks or months of data collection.

Key Responsibilities:

• Deploy Pytorch models on Nvidia Jetson platforms using optimal TensorRT optimizations.
• Interface off-the-shelf hardware accelerators with single board computers like Orange Pi and Raspberry Pi.
• Interact with various hardware accelerators (e.g. GPUs), debug issues, and optimize C++ code for maximum performance.
• Troubleshoot power draw issues from SSDs, USB cameras, AI boards, CPUs, and GPUs.
• Investigate the support for different machine learning operations on various compute platforms.

Requirements:

• Graduate student with a background in Electrical Engineering focused on Machine Learning/Deep Learning for Computer Vision, or, Undergraduate students with relevant experience.
• High proficiency in C++ with hands-on experience in embedded Linux.
• Experience with writing and deploying machine learning algorithms.
• Strong understanding of PCIE interfaces for NVMEs, HW accelerators, and WiFi cards.
• Solid knowledge of ML concepts like convolutions, encoders, decoders, optimizers, loss functions, and their implementation on embedded platforms.
• Experience working with and debugging the full Linux stack system.
• Experience/familiarity with Nvidia Jetson platforms and understanding of their HW components (tensor cores, DLA, video encoders & decoders, etc.).
• Experience with various digital interfaces (I2C, SPI, USB, CAN, HDMI, DDR3/4).
• Familiarity with scripting languages like Python or Bash.
• Experience working with arm64-based platforms.

Compensation:

The estimated hourly pay guideline range for this role is between $30.00 – $45.00 and may be modified based on market and individual qualifications.

Equal Opportunity Employer:

Invisible AI is an equal opportunity employer and does not discriminate based on age, ethnicity, gender, nationality, religious belief, or sexual orientation.