Senior AI Engineer - Machine Learning (US)

Gauss Labs; Palo Alto (CA); Hybrid; USD 180,000 - 260,000; Full time

Job summary

Gauss Labs in Palo Alto, CA, is seeking a Senior/Staff AI Engineer to turn ML research into robust production systems, owning them end-to-end across lifecycle.

You will collaborate with AI Scientists, Software Engineers, and PMs across Palo Alto and Seoul to scale tabular foundation models and deploy them at large scale with reliable performance.

Qualifications

• BS in Computer Science, Electrical Engineering, Machine Learning, or related field; plus6+ years of full-time experience or MS/PhD with 4+ years of full-time experience.
• Strong Python programming with deep ML stack experience (NumPy, Pandas, scikit-learn, PyTorch or TensorFlow).
• 3+ years building production-grade ML infrastructure — data pipelines, training/inference workflows, deployment automation, CI/CD, and containerization (Docker/Kubernetes).
• Track record of shipping ML systems with focus on scalability, performance, and reliability.
• Demonstrated leadership: owning roadmaps for multi-quarter, multi-team efforts and mentoring engineers.

Responsibilities

• Partner with AI Scientists to build and productionize tabular foundation models from pretraining to serving.
• Build reliable ML infrastructure across data, training, inference pipelines, CI/CD, observability, and scalable workflows.
• Design evaluation/monitoring to reflect real-world usage, handling distribution shift and limited labels.
• Set engineering standards, lead design reviews, and drive adoption of new modeling approaches and infra.
• Partner with product and engineering teams to integrate ML into user-facing systems.
• Define scope/roadmaps for multi-team initiatives and mentor senior engineers.

Skills

Python

Algorithms

Data structures

NumPy

Pandas

scikit-learn

PyTorch

TensorFlow

Git

CI/CD

Docker

Kubernetes

Model deployment

Mentoring

Education

BS in CS/EE/ML

MS/PhD in related field

Tools

Docker

Kubernetes

Git

CI/CD

PyTorch

TensorFlow

NumPy

Pandas

scikit-learn

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, plus6+ yearsof full-time experience;orMS/PhD plus4+ yearsof 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, plus8+ yearsof full-time experience;orMS/PhD plus6+ yearsof 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.