Kairos
Back to jobs

Senior Applied AI Engineer - Multimodal Transformers

On-site
Kodiak AISan Francisco Bay Area4 months agoWebsite
Fresh
Senior
AI & Machine Learning

Compensation

$200,000-$260,000
Apply
Share

Description

Kodiak's autonomy stack is built on AI that fuses diverse sensor streams into a unified, actionable understanding of the world. We are developing GigaFusionNet – a large-scale multimodal transformer that learns rich, joint representations across camera, LiDAR, and radar through attention-based fusion. We are looking for engineers to push the boundaries of how transformer architectures combine and reason over heterogeneous sensor data.This role is open to all levels – from those eager to contribute to cutting-edge research to experts driving innovation at scale.

In this role, you will:
  • Design and develop multimodal transformer architectures that fuse camera, LiDAR, and radar into unified representations
  • Research and implement cross-modal attention mechanisms, token fusion strategies, and efficient multi-stream tokenization
  • Build scalable training pipelines for large-scale multimodal transformers across massive real-world datasets
  • Explore self-supervised and contrastive pretraining objectives that learn transferable multimodal representations
  • Optimize transformer models for real-time inference under latency and compute constraints
What you’ll bring:
  • BS, MS, or PhD in AI, Computer Science, or a related field
  • 4+ years experience with transformer architectures, particularly in multimodal or multi-stream settings
  • Familiarity with cross-attention, token fusion, or modality alignment techniques
  • Proficiency in Python and deep learning frameworks like PyTorch or TensorFlow
  • Strong understanding of scalable training for large models, including distributed training and mixed-precision optimization
  • Passion for building AI that reasons over the full breadth of sensory input to operate safely in the real world

What we offer:

  • Competitive compensation package including equity and annual bonuses
  • Excellent Medical, Dental, and Vision plans through Kaiser Permanente, Cigna, and  MetLife (including a medical plan with infertility benefits)
  • MetLife Legal Services, Identity & Fraud Protection, Hospital Indemnity Insurance, Accident Insurance, & Critical Illness Insurance
  • Flexible PTO, 10 paid holidays, and generous parental leave policies
  • Our office is centrally located in Mountain View, CA
  • Office perks: dog-friendly, free catered lunch, a fully stocked kitchen, and free EV charging
  • Long Term Disability, Short Term Disability, Life Insurance
  • Wellbeing Benefits - Headspace through Cigna, Calm through Kaiser, One Medical, Gympass, Spring Health through Cigna, Rula (mental health navigation) 
  • Fidelity 401(k)
  • Commuter, FSA, Dependent Care FSA, HSA
  • Various incentive programs (referral bonuses, patent bonuses, etc.)

Stack

PythonPyTorchTransformersTensorFlowDeep Learning
Posted
Feb 20, 2026
Last seen
Jun 25, 2026
First seen
Jun 25, 2026
Status
active