Staff ML Engineer - Infrastructure jobs in United States
cer-icon
Apply on Employer Site
company-logo

ChipStack ยท 6 months ago

Staff ML Engineer - Infrastructure

positionSan Jose
timeFull-time
remoteOnsite
seniorityMid, Senior Level
date5+ years exp

ChipStack is a startup focused on reinventing the design of modern silicon chips, leveraging expertise in AI and systems software. They are seeking a strong ML Infrastructure Engineer to join their founding team, responsible for building infrastructure that supports the training and deployment of large language models in various environments.

Artificial Intelligence (AI)Information TechnologySemiconductor
check
H1B Sponsor Likelynote

Responsibilities

Building the core infrastructure that enables training, fine-tuning, evaluation, and deployment of LLMs across cloud and on-premise environments

Qualification

ML infrastructure designPythonPyTorchTensorFlowCloud platformsDistributed trainingData pipelinesPerformance optimizationDeploying ML modelsContainerized systemsGPU/TPU workload managementTraining LLMsMonitoringChip design fundamentalsEarly-stage startup experienceCommunication skillsObservability

Required

5+ years of experience in ML infrastructure or adjacent roles
Deep expertise in Python and experience with training frameworks like PyTorch or TensorFlow
Strong systems engineering skills and experience with distributed training, data pipelines, and performance optimization
Experience deploying ML models to production (REST APIs, batch jobs, streaming pipelines)
Proficiency with cloud platforms (e.g., GCP, AWS) and containerized systems (Docker, Kubernetes)
Experience managing GPU/TPU workloads efficiently
Good communication skills and the ability to work directly with engineers and customers
Prior experience training or fine-tuning LLMs
Experience setting up observability, monitoring, and evaluation pipelines for ML models

Preferred

Exposure to chip design fundamentals (via coursework or elsewhere)
Experience at an early-stage startup

Company

ChipStack

twittertwittertwitter
company-logo
Chipstack utilizes generative AI-driven tools to reimagine chip design.
location
Seattle, Washington, USA
people-size11-50 employees
web-link
https://www.chipstack.ai

H1B Sponsorship

ChipStack 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 (2)
2024 (1)

Funding

Current Stage
Early Stage
Total Funding
unknown
Key Investors
Khosla Ventures
2025-11-10Acquired
2025-01-01Seed

Recent News

Techgist Africa | Africa Leading Tech News, Reviews and Tips
Cadence Design Systems Acquires ChipStack to Boost AI-Driven Verification Capabilities
2025-11-14
GeekWire
Cadence to acquire Seattle startup ChipStack to boost chip design automation
2025-11-13
Company data provided by crunchbase