GenAI MLOps Engineer — AI Trainer (JAX & Kernels)

Obsidian; San Francisco (CA); On-site; USD 120,000 - 160,000; Full time

Job summary

Obsidian is looking for an experienced MLOps Engineer to join their innovative GenAI team in San Francisco. You will work on building foundational AI models, engaging in AI model training and evaluation, and developing solutions for high-quality training data generation.

Ideal candidates have extensive experience with JAX and kernel-level programming and strong communication skills. This is a full-time position, requiring a reliable engagement of 40 hours per week.

Qualifications

• 2+ years of dedicated professional experience in ML infrastructure, MLOps, or ML systems engineering at a recognized organization.
• Hands-on production experience with JAX at scale.
• Experience writing or optimizing custom GPU kernels using Pallas or Triton.

Responsibilities

• Guide research and engineering teams to improve AI model performance.
• Design challenging, domain-relevant tasks for MLOps solutions.
• Evaluate MLOps tasks and provide clear, written technical feedback.

Skills

MLOps

JAX

Kernel-level programming

Production experience

Strong written communication

Tools

Pallas

Triton

Job description

Obsidian is looking for an experienced MLOps Engineer to join their innovative GenAI team in San Francisco. You will work on building foundational AI models, engaging in AI model training and evaluation, and developing solutions for high-quality training data generation.

Ideal candidates have extensive experience with JAX and kernel-level programming and strong communication skills. This is a full-time position, requiring a reliable engagement of 40 hours per week.