GenAI MLOps Engineer — Build Large-Scale AI Models

Obsidian; New York (NY); Remote; USD 100,000 - 130,000; Full time

Job summary

Obsidian is seeking a talented MLOps Engineer to join their cutting-edge GenAI team in New York. This full-time role involves training and evaluating AI models, developing solutions to MLOps problems, and ensuring the accuracy of training data.

Candidates should have over 2 years of experience in ML infrastructure or MLOps, hands-on experience with JAX, and a strong ability to communicate complex technical concepts. Commitment to a 40-hour work week is required.

Qualifications

• 2+ years of experience in ML infrastructure, MLOps, or ML systems engineering.
• Hands-on experience with JAX at scale.
• Experience with custom GPU kernel optimization.

Responsibilities

• Guide teams to improve AI model performance in MLOps.
• Design domain-relevant tasks and write accurate solutions.
• Evaluate MLOps tasks and provide technical feedback.
• Develop evaluation frameworks for training pipeline design.
• Collaborate with experts for accuracy in training data.

Skills

JAX

Kernel-level programming

MLOps

GPU kernels (Pallas/Triton)

Written communication

Job description

Obsidian is seeking a talented MLOps Engineer to join their cutting-edge GenAI team in New York. This full-time role involves training and evaluating AI models, developing solutions to MLOps problems, and ensuring the accuracy of training data.

Candidates should have over 2 years of experience in ML infrastructure or MLOps, hands-on experience with JAX, and a strong ability to communicate complex technical concepts. Commitment to a 40-hour work week is required.