Software Engineer, Machine Learning Infrastructure jobs in United States
cer-icon
Apply on Employer Site
company-logo

David AI · 4 months ago

Software Engineer, Machine Learning Infrastructure

David AI is an innovative company focused on transforming raw audio into high-signal data for AI labs and enterprises. As a Software Engineer in Machine Learning Infrastructure, you will build and scale the core infrastructure for audio ML products, enabling efficient training, deployment, and evaluation of machine learning models.

Artificial Intelligence (AI)AudioGenerative AIMachine Learning
check
H1B Sponsor Likelynote

Responsibilities

Design and maintain data pipelines for processing massive audio datasets, ensuring terabytes of data are managed, versioned, and fed into model training efficiently
Develop frameworks for training audio models on compute clusters, managing cloud resources, optimizing GPU utilization, and improving experiment reproducibility
Create robust infrastructure for deploying ML models to production, including APIs, microservices, model serving frameworks, and real-time performance monitoring
Apply software engineering best practices with monitoring, logging, and alerting to guarantee high availability and fault-tolerant production workloads
Translate research prototypes into production pipelines, working with ML engineers and data teams to support efficient data labeling and preparation
Evaluate and integrate new MLOps technologies and optimization techniques to enhance infrastructure velocity and reliability

Qualification

Machine Learning InfrastructureCloud InfrastructureData PipelinesPythonDockerKubernetesCI/CD PipelinesSystem DesignExperiment TrackingFast-paced Environment

Required

5+ years of backend engineering with 2+ years ML infrastructure experience
Hands-on experience scaling cloud infrastructure and large-scale data processing pipelines for ML model training and evaluation
Proficient with Docker, Kubernetes, and CI/CD pipelines
Proven ML model deployment and lifecycle management in production
Strong system design skills optimizing for scale and performance
Proficient in Python with deep Kubernetes experience

Preferred

Experience with feature stores, experiment tracking (MLflow, Weights and Biases), or custom CI/CD pipelines
Familiarity with large-scale data ingestion and streaming systems (Spark, Kafka, Airflow)
Proven ability to thrive in fast-moving startup environments

Benefits

Rapid career growth at one of the fastest growing Series A companies, within a new and booming industry.
Competitive salary and equity package.
Flexible PTO policy.
Top-notch health, dental, and vision coverage with 100% company reimbursement for most plans.
Paid lunch and dinner in the office, every day through DoorDash.
401k access.

Company

David AI

twittertwittertwitter
company-logo
David AI supplies audio datasets and associated services for speech, multilingual, and conversational AI applications.

H1B Sponsorship

David AI 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)

Funding

Current Stage
Growth Stage
Total Funding
$80.5M
Key Investors
First Round CapitalY Combinator
2025-10-08Series B· $50M
2025-05-22Series A· $25M
2025-01-16Seed· $5M

Recent News

Tech Startups - Startups and Technology news
Company data provided by crunchbase