Senior AI Engineer - Machine Learning (US)
Gauss Labs
Apply to this job Palo Alto, CA hybrid Until 8/21/2026 4+ years exp H-1B sponsor history First posted June 21, 2025 Last posted June 21, 2025
Job description
Gauss Labs builds Industrial AI for the world's leading manufacturers, applying state-of-the-art ML to large volumes of real production data. A core focus of this role is building our internal tabular foundation models — pretraining, fine-tuning, and serving them in large-scale production systems. As a Senior/Staff AI Engineer, you will turn ML research into robust, scalable production systems and own them across their full lifecycle. You'll work with AI Scientists, Software Engineers, and Program Managers across Palo Alto, CA, and Seoul, South Korea.
Responsibilities
- Partner with AI Scientists to build and productionize our tabular foundation models — owning the pretraining, fine-tuning, serving, and monitoring infrastructure and scaling it from research prototype to large-scale production.
- Build reliable, performant ML infrastructure across research, staging, and production: data, training, and inference pipelines, CI/CD, observability, and reproducible workflows, tuned for latency, throughput, and resource usage.
- Design evaluation and monitoring that reflect how models are actually used, including handling real-world data challenges such as distribution shift and limited labels.
- Set engineering standards, lead design and architecture reviews, and drive adoption of new modeling approaches, algorithms, and infrastructure.
- Partner with product and engineering teams to integrate ML into user-facing systems.
- Define scope and roadmap for multi-team initiatives, drive cross-team and cross-functional alignment across AI Science, engineering, and product, and mentor senior engineers to raise the organization's technical bar.
Key Qualifications
- BS in Computer Science, Electrical Engineering, Machine Learning, or a related technical field, plus 6+ years of full-time experience; or MS/PhD plus 4+ years of full-time experience.
- Strong programming skills in Python with a solid grounding in algorithms and data structures, and deep proficiency with the Python data/ML stack (NumPy, Pandas, scikit-learn, and PyTorch or TensorFlow) for end-to-end model development.
- 3+ years building production-grade ML infrastructure — data pipelines, training/inference workflows, and deployment automation — with solid software engineering fundamentals (Git, testing, code review, CI/CD, and containerization/orchestration such as Docker and Kubernetes).
- Track record of shipping ML systems with real attention to scalability, performance, and reliability.
- Demonstrated technical leadership: owning the roadmap for multi-quarter, multi-team efforts, driving cross-team and cross-functional alignment, and mentoring other engineers.
Preferred Qualifications
- BS in Computer Science, Electrical Engineering, Machine Learning, or a related technical field, plus 8+ years of full-time experience; or MS/PhD plus 6+ years of full-time experience.
- Experience pretraining, fine-tuning, and serving foundation models and transformers of any modality (tabular, timeseries, language, vision, etc.) at scale.
- Experience optimizing training and inference for large-scale models, including distributed/parallel training (multi-GPU/multi-node).
- Experience deploying ML in production across batch, real-time, or edge settings.
- Experience with models under real-world data challenges — distribution shift, limited or noisy labels, or continual learning.
- Development in a cloud environment (AWS, Azure, or GCP).
- Productive with AI-assisted / agentic coding tools (e.g. Claude Code, Copilot) and eager to push their limits — building workflows and agents that raise the team's velocity.
About this role
Summary
Build scalable ML systems, manage lifecycle, and lead cross-team AI projects.
Job title
Senior AI Engineer - Machine Learning (US)
Experience level
6+ years or 4+ years with MS/PhD
Minimum experience
4+ years exp
Industry
software
Location requirements
Palo Alto, CA; remote work not specified.
Salary
Not specified
Visa sponsorship
H-1B sponsor history
Management role
No
Skills & keywords
Required skills
pythonalgorithmsdata structuresnumpypandasscikit-learnpytorchtensorflowgittestingci/cddockerkubernetes
Preferred skills
foundation modelstransformersdistributed trainingmulti-GPUcloud computingawsgcplinuxedge deploymentcontinual learningagentic coding
Specializations
machine learningdeep learningmodel deploymentML infrastructure
Locations
Structured locations inferred from the posting.
Palo Alto, CA, USA
Hybrid City
Related searches