Labs Companies
Posted 3mo ago

AI Architect — Model Design, Training & Multi-Modal Deployment

Labs Companies
Santa Monica, California, United States
$240k-$340k/yrOnsiteFull Time
Responsibilities
  • designing models
  • training systems
  • mentoring engineers
Requirements
  • Requires 7+ years in ML engineering
  • 3+ years training or fine-tuning large models
  • Deep PyTorch or JAX expertise
  • Distributed training
  • Production inference at scale, and model evaluation skills
Technical tools mentioned
PyTorchJAXCUDATritonTransformersRLHFvLLMRayKubernetesFSDPDeepSpeedTPU

Job description

Overview

Design, train, and deploy the multi-modal models behind Labs Intelligence — the AI substrate for every Labs product, from Yuhmmy's taste intelligence to XR perception.

Mission

The 7 Frontier Intelligences begin with Superintelligence. This role is the architect of that beginning — models that perceive, reason, and act across modalities in service of human enhancement.

What you'll do

  • Own model architecture decisions across text, vision, audio, and sensor modalities
  • Build and operate the training stack — data pipelines, distributed runs, evaluation harnesses
  • Design the deployment path: quantization, serving (vLLM/Triton), latency and cost budgets
  • Stand up RLHF/preference-tuning loops with product feedback in the loop
  • Set research-to-production standards for every Labs team consuming models
  • Mentor engineers and researchers; review the hard technical calls

Minimum qualifications

  • 7+ years in ML engineering with 3+ training or fine-tuning large models
  • Deep PyTorch or JAX; real distributed-training experience (FSDP/DeepSpeed/TPU or equivalent)
  • Production inference at scale — you have owned a serving fleet and its bill
  • Strong evaluation instincts: you measure before and after you believe

Preferred qualifications

  • Multimodal or embodied-AI research record
  • CUDA/Triton kernel work
  • RLHF pipelines in production

Nice to have

  • Open-source contributions to the ML stack
  • Publications or patents
  • Edge/on-device deployment

Tech stack

PyTorchJAXCUDATritonTransformersRLHFvLLMRayKubernetes

Growth

Define the AI organization of a Supercompany — this seat becomes Labs' senior-most technical AI leadership.

Impact

Every Labs product inherits your architecture. One role, seven intelligences, an entire ecosystem's ceiling.

Projects you'll touch

  • Labs Intelligence foundation models
  • Multi-modal perception stack
  • Yuhmmy AI + XR model services

Benefits

  • Meaningful early equity in a $12.5B frontier company
  • Premium medical, dental, and vision for you and your family
  • Unlimited PTO with a 3-week minimum, plus Labs Recharge weeks
  • 401(k) with company match
  • Relocation support to Santa Monica
  • Daily chef-crafted (vegan-friendly) meals — taste is our business
  • Annual learning budget + Labs University access
  • Latest hardware and an XR/device lab at your desk
  • Parental leave: 20 weeks, all parents
  • Yuhmmy family access and Labs brand perks

Labs Companies, Inc. is an equal-opportunity employer. Posted May 12, 2026 · Req D32799

About Labs Companies

Developing human-enhancement technologies and intelligent ecosystems.

Similar jobs

AI Architect roles near Santa Monica, California
1w
Save
Mark Applied
Hide
Secure AI Architect Senior Manager/Assoc Director
Albany or Arlington or Atlanta or Austin or Beaverton or Bentonville or Boston or Carmel or Charlotte or Chicago or Cincinnati or Cleveland or Columbus or Culver City or Denver or Des Moines or Detroit or Hartford or Houston or Irving or Kirkland or Miami or Milwaukee or Minneapolis or Morristown or Mountain View or Nashville or New York City or Oklahoma City or Overland Park or Philadelphia or Pittsburgh or Raleigh or Redmond or Sacramento or San Diego or San Francisco or Scottsdale or Seattle or St. Louis or St. Petersburg or Walnut Creek or United States
$113k-$338k/yr HybridFull Time
Accenture
AccentureNYSE: ACN: Global professional services firm providing consulting and technology solutions.
8+ YOERequires 8+ years in cybersecurity, 1+ year in Secure AI, agentic coding, AI architecture, AI security, and governance frameworks, plus a bachelor's degree or equivalent experience.
Anthropic Claude Code, Cursor, GitHub Copilot, Codex, RAG, MLOps, A2A, MCP Gateways, Anthropic Claude, OpenAI, Azure OpenAI, OWASP LLM Top 10, EU AI Act, NIST AI RMF
4w
Save
Mark Applied
Hide
Agentic AI Architect-Google Cloud 
Boston or Chicago or Dallas or Los Angeles or Minneapolis or New York City or San Francisco or Seattle or Washington
$163k-$221k/yr OnsiteFull Time
West Monroe
West Monroe: Business and technology consultancy specializing in digital transformation.
8+ YOE8+ years in software/cloud/platform engineering, 3+ years on Google Cloud, hands-on experience with agentic AI and platform engineering, strong software engineering skills, GCP certifications preferred, ability to travel 25–50%, and authorized to work in the US.
Gemini, Agent Development Kit (ADK), Agent Engine, Vertex AI, Vertex AI Search, Vector Search, Model Context Protocol (MCP), Google Kubernetes Engine (GKE), Cloud Run, Cloud Functions, Pub/Sub, BigQuery, Cloud SQL, Kubernetes, Docker, Terraform, GitHub Actions, Cloud Build, Java, Python, Go, TypeScript
1mo
Save
Mark Applied
Hide
Agentic AI Architect - Intelligence Engineering (West)
Denver or East Bay or Los Angeles or Orange County or Phoenix or Portland or Salt Lake City or San Diego or San Francisco or Seattle or Silicon Valley
$108k-$237k/yr HybridFull Time
Slalom
Slalom: Provides business and technology consulting and software engineering services.
5+ YOE5+ years software engineering experience with 2+ years on generative/agentic AI, multi-agent system design, RAG architectures, cloud AI platform deployment, Python proficiency, and strong communication skills.
Python, FastAPI, Flask, Lang Graph, Strands Agents SDK, OpenAI Agents SDK, Google ADK, LangSmith, Model Context Protocol (MCP), AWS Bedrock, Azure AI Foundry, OpenAI Service, GCP Vertex AI, Gemini, Databricks, Claude Code, Cursor, Kiro, RAGAS, DeepEval
2mo
Save
Mark Applied
Hide
AI - Architect
Los Angeles or Nashville or New York City or Memphis or Chicago or Miami or London or Munich or Geneva or Stockholm or Shanghai or Beijing
OnsiteFull Time
Creative Artists Agency
Creative Artists Agency: Provides talent representation and brand consulting services globally.
10+ YOE10+ years building AI/ML/NLP solutions, 5+ years building applied AI/LLM systems in production; Bachelor's required (Master's preferred); strong Python and software engineering skills; experience with data, retrieval, and production architectures.
Python
5mo
Save
Mark Applied
Hide
Agentic AI Architect
Irvine, California, United States
$159k-$270k/yr OnsiteFull Time
Ingram Micro
Ingram MicroNYSE: INGM: Global distributor of technology products and supply chain solutions.
10+ YOEMaster’s or Ph.D. in a technical field, 10+ years in software engineering and AI/ML, agentic AI architecture expertise, Python, distributed systems, cloud platforms, data architecture, and MLOps/LLMOps.
Google ADK, LangChain, AutoGen, CrewAI, ERP, CRM, SCM, WMS, LLMs, LlamaIndex, Microsoft Semantic Kernel, Vertex AI Search and Conversation, Agent Builder, GPT, Claude, Gemini, Llama, Python, TensorFlow, PyTorch, scikit-learn, Hugging Face Transformers, MLOps, LLMOps, APIs, vector databases, knowledge graphs, GCP, AWS, Azure, reinforcement learning (RL), Architecture Review Boards
4w
Save
Mark Applied
Hide
Head of AI Workflows / Lead Architect
London or New York City or Los Angeles
$150k-$200k/yr HybridFull Time
Wonder Studios
Wonder Studios: AI-native studio producing premium film and creative content.
Significant VFX/post-production or production engineering experience; hands-on knowledge of generative and VFX image/video systems; track record taking prototypes to production; strong communication and product judgment.
ComfyUI, Stable Diffusion, Flux, Runway, Veo, Luma, Adobe Creative Cloud, DaVinci Resolve, Unreal Engine, Nuke, Maya
1mo
Save
Mark Applied
Hide
VP, AI Solutions Architect
New York City or Chicago or Detroit or Los Angeles
$163k-$263k/yr HybridFull Time
Dentsu
DentsuTokyo Stock Exchange: 4324: Providing global advertising, digital marketing, and media communications services.
12+ YOE12+ years in solution/enterprise architecture or technical consulting with 3+ years delivering production generative AI/LLM systems, hands-on agentic system experience, cloud-native architecture skills, programming proficiency, and strong client-facing leadership.
LangGraph, Microsoft Agent Framework, OpenAI Agents SDK, Claude Agent SDK, CrewAI, LangSmith, Arize, Braintrust, Salesforce, Adobe, Microsoft, AWS, Google Cloud, Databricks, Snowflake, Python, Java, C#, TypeScript, JavaScript, Microsoft Foundry, AWS Bedrock, AgentCore, Google Gemini Enterprise, Vertex AI, Adobe Experience Cloud, AEM, AEP, GenStudio, AEP Agent Orchestrator, Salesforce Agentforce, Data Cloud, Braze, Shopify, Marketplacer, PyTorch
1mo
Save
Mark Applied
Hide
AI & Data Architect - Trust and Safety
Los Angeles or Singapore or New York City or London or Dublin or Paris or Berlin or Dubai or Jakarta or Seoul or Tokyo
$219k-$408k/yr OnsiteFull Time
TikTok
TikTok: Global short-form video hosting and social media platform.
5+ YOE5+ years building data and AI architectures, experience with scalable data ecosystems, GenAI patterns (RAG, embeddings), and translating complex safety problems into production solutions.
RAG, embeddings, vector databases, knowledge graphs, MLOps