Solidigm · 1 hour ago
AI Software Engineer
Solidigm is a multibillion-dollar global company in the memory industry, seeking an AI Software Engineer to join their innovative team. The role involves designing and developing deep learning software, optimizing frameworks, and enabling machine learning operations for AI productization.
Data CenterData StorageHardware
Responsibilities
Conduct design and development to build and optimize deep learning software
Design, develop & optimize for deep learning training and inference frameworks
Implement various distributed algorithms such as model/data parallel frameworks, parameter servers, dataflow based asynchronous data communication in deep learning frameworks
Transform computational graph representation of neural network model
Develop deep learning primitives in math libraries
Profile distributed DL models to identify performance bottlenecks and propose solutions across individual component teams
Optimizing code for various computing hardware backends
Interacting with deep learning researchers and experience with deep learning frameworks
Responsible for preparing data for ML models at scale, building appropriate inference interfaces for ML model consumption, enabling ML Ops for continuous delivery and automation of ML pipelines, and/or building and sustaining AI productization platforms
Qualification
Required
MS degree with 6 years of experience, or a PhD with up to 2 years experience
6-8 years of experience in AI, and Data Science
Hands-on model development and up to date with AI technologies
Experience developing in Python
PyTorch and/or Tensorflow
Focus on computer vision use cases and models
Preferred
Direct work and interactions with customers
Robotics and physical AI
Company
Solidigm
Solidigm provides storage products and solutions for data centers.
H1B Sponsorship
Solidigm 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 (66)
2024 (70)
2023 (58)
2022 (48)
Funding
Current Stage
Late StageRecent News
2025-11-21
2025-11-19
Company data provided by crunchbase