Principal AI Architect & Hands-On Technical Lead

Location: Dublin, CA Salary: $185,000.00 USD Annually - $210,000.00 USD Annually Description: Principal AI Architect & Technical Lead

Location: Pleasanton, CA (Hybrid)

Salary: 185,000-$210,000

About the Opportunity

We are seeking a Principal AI Architect & Technical Lead to architect, build, and lead full-lifecycle AI/ML product development, from data engineering and model training through GenAI integration and intelligent application deployment.

This is an end-to-end leadership role for a hands-on builder who thrives on writing production code, deploying machine learning systems, integrating large language models (LLMs), and leading high-performing teams of AI and software engineers. The ideal candidate combines deep technical expertise with strategic leadership to deliver innovative, scalable AI solutions.
What You'll Do
Architect & Build AI Solutions

• Design and develop end-to-end machine learning systems, including data pipelines, feature stores, training environments, inference services, and monitoring frameworks.
• Architect scalable, cloud-native applications that leverage foundation models, Retrieval-Augmented Generation (RAG), and agentic AI workflows.
• Develop secure, privacy-focused machine learning solutions aligned with industry compliance and data governance requirements.
• Create explainable AI frameworks that provide transparency and trust in model outputs.
• Design multi-tenant AI architectures that ensure data isolation, scalability, performance, and reliability.

Hands-On Engineering

• Build and maintain robust data pipelines using BigQuery, Dataflow, Apache Beam, and related technologies.
• Train, optimize, and deploy machine learning models using TensorFlow, PyTorch, and Scikit-learn.
• Develop full-stack AI-powered applications using Python, Java, React, Angular, and cloud-native services.
• Implement MLOps practices that support reproducible training, CI/CD automation, model monitoring, and cost-effective scaling.

Generative AI & Agent Development

• Build and integrate GenAI capabilities including semantic search, knowledge retrieval, intelligent Q&A, summarization, conversational AI, and synthetic data generation.
• Design and develop AI agents and copilots using tools such as LangChain, CrewAI, Vertex AI, and related frameworks.
• Develop domain-specific LLM applications leveraging vector databases including FAISS and Pinecone.
• Prototype and deploy multi-agent systems that streamline complex business processes and decision-making workflows.

Technical Leadership

• Lead and mentor a team of AI engineers, machine learning engineers, and software developers.
• Define technical vision, architecture standards, and engineering best practices.
• Partner with Product, Data Science, and Executive Leadership teams to drive AI strategy and innovation.
• Guide architectural decisions and ensure successful delivery of AI initiatives from concept to production.

Required Qualifications

• 10+ years of experience in software engineering, machine learning engineering, or related fields.
• 5+ years of experience leading engineering teams and technical architecture initiatives.
• Proven success designing, building, and deploying production-scale AI/ML applications.
• Deep expertise with Generative AI technologies, LLM frameworks, agentic AI architectures, and MLOps practices.
• Experience operationalizing enterprise AI solutions, including governance, security, monitoring, and cost optimization.
• Demonstrated track record of delivering innovative products at scale.
• Strong communication, leadership, and cross-functional collaboration skills.

Preferred Technical Skills
Programming Languages & Frameworks

• Python
• Java
• SQL
• JavaScript / TypeScript
• TensorFlow
• PyTorch
• Scikit-learn
• React
• Angular

Cloud Platforms

• Google Cloud Platform (Google Cloud Platform)
  • Vertex AI
  • BigQuery
  • Cloud Functions
  • Cloud Run
  • Dataflow
  • Pub/Sub
• Oracle Cloud Infrastructure (OCI)
  • OCI Data Science
  • Autonomous Data Warehouse
  • OCI Functions
  • Oracle Kubernetes Engine (OKE)
  • Data Flow
  • Streaming

Data & MLOps

• BigQuery
• AlloyDB
• MongoDB
• Redis
• MLflow
• Kubeflow
• Apache Airflow
• Docker
• Kubernetes
• GitHub Actions
• Terraform

Generative AI & Agent Frameworks

• Vertex AI Gemini
• OpenAI
• Claude
• LangChain
• LlamaIndex
• FAISS
• Pinecone
• RAG Architectures
• Multi-Agent Systems
• Prompt Engineering & Orchestration

Developer Tools

• GitHub Copilot
• ChatGPT
• Cursor AI
• Locofy

Competitive compensation packages and benefits depending on location and position,

• Comprehensive medical, dental, and vision insurance
• 401(k) retirement savings plan
• Paid time off and company-recognized holidays
• Professional development and career growth opportunities
• Collaborative, innovative, and entrepreneurial work environment
• Exposure to cutting-edge AI technologies and high-impact initiatives
• Opportunity to make a meaningful contribution to a growing organization

If you're passionate about building transformative AI products and leading world-class engineering teams, we'd love to hear from you.

Please send resume and Linked IN to

Medical, dental, and vision insurance are available to qualified candidates who meet eligibility requirements.
By providing your phone number, you consent to: (1) receive automated text messages and calls from the Judge Group, Inc. and its affiliates (collectively "Judge") to such phone number regarding job opportunities, your job application, and for other related purposes. Message & data rates apply and message frequency may vary. Consistent with Judge's Privacy Policy, information obtained from your consent will not be shared with third parties for marketing/promotional purposes. Reply STOP to opt out of receiving telephone calls and text messages from Judge and HELP for help.
Contact:
This job and many more are available through The Judge Group. Please apply with us today!