Sr. Software Engineer

We are seeking a highly skilled and motivated embedded software Engineer to join our team and optimize various AI models for CVflow.

The ideal candidate will possess expertise in neural networks, hardware architecture, and embedded programming. You will be responsible to build and deploy state of the art deep learning models onto the Ambarella SoC.

Responsibilities:

• PyTorch and AI Expertise:
• Train model for quantization and pruning. Understand how backpropagation of training.
• Able to verify models for accuracy.
• Able to modify pytorch code to model Ambarella SoC.
• Create scripts and utilities for data preprocessing, model training, validation and inference.
• Stay up-to date with new and emerging model architectures and integrate them into the model garden.
• Performance Analysis and Optimization:
• Profile and analyze the performance of PyTorch on target hardware.
• Identify and address performance bottlenecks. Verify performance for speed and memory bandwidth.
• Develop and implement optimization techniques to improve training and inference speed.
• Model Garden:
• Work closely with developers and users of model garden to help troubleshoot issues related to using the models.
• Develop comprehensive tutorials and examples to use model garden effectively.

Software Engineering, Software Engineering, Embedded Software, Artificial Intelligence (AI), Neural Networks, Hardware Architecture, Deep Learning, Embedded Programming Languages, System-on-a-Chip (SoC), Network System Hardware, * Required:

• BS in Computer Science, Electrical Engineering, or a related field.
• Developing or using deep learning frameworks (e.g., Tensorflow, PyTorch).
• Knowledge of ML lifecycle - data preprocessing, training, deployment and inference,
• deep learning and its applications to computer vision (e.g., classification, object detection, and segmentation);
• Experience with specific hardware architectures (e.g., NVIDIA GPUs, custom ASICs).
• Strong C/C++ and Python programming skills.
• Modern software development methodologies: source control, code review, unit testing, and peer reviews.
• Team player with strong communication skills.
• Preferred:
• MS in Computer Science or related fields.
• Industry experience in related field.
• Familiarity with compression techniques like QAT, pruning (nice to have)
• Containerization techniques like docker
• Deep understanding of hardware architectures, including GPUs, ASICs, and other accelerators.

The base salary range is $161,000 - $182,000. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The successful candidate will have the opportunity to convert to a full-time regular position. We also offer new-hire RSU grants and the opportunity for annual RSU grants, as well as other highly competitive benefits.