Ambarella Inc · 4 days ago
Software Engineer II
Ambarella Inc is seeking a highly skilled embedded software engineer to optimize various AI models for CVflow. The role involves building and deploying deep learning models onto the Ambarella SoC and requires expertise in neural networks and embedded programming.
Responsibilities
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
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
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
Qualification
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
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
Company
Ambarella Inc
Ambarella (Nasdaq: AMBA) is a leading developer of visual AI products.
H1B Sponsorship
Ambarella Inc has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (31)
2024 (21)
2023 (19)
2022 (17)
2021 (38)
2020 (19)
Funding
Current Stage
Public CompanyTotal Funding
$38.5MKey Investors
Riverwood Capital
2012-10-10IPO
2012-01-01Private Equity
2006-02-01Series C· $15.4M
Recent News
thefly.com
2026-01-08
2026-01-06
Company data provided by crunchbase