AI Software Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

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
check
H1B Sponsor Likelynote

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

Deep LearningPythonPyTorchTensorFlowAI TechnologiesData ScienceComputer VisionRoboticsCustomer Interaction

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

twittertwittertwitter
company-logo
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 Stage

Leadership Team

leader-logo
Alan Bumgarner
Director of Strategic Planning
linkedin
Company data provided by crunchbase