MLOps Engineer for AI Model Training

Be at the forefront of the AI revolution by joining a leading AI lab''s GenAI team, where your expertise will drive the development of advanced Large Language Models. This role is pivotal in building foundational AI models and requires MLOps Engineers with extensive hands-on experience in modern ML frameworks, particularly JAX, PyTorch, and kernel-level programming (Pallas/Triton). Key Responsibilities

• Guide research and engineering teams to bridge knowledge gaps and enhance AI model performance in MLOps, training infrastructure, and ML framework-level topics.
• Design complex, domain-relevant tasks and produce accurate, well-structured solutions to MLOps and ML systems challenges.
• Evaluate MLOps tasks and solutions, providing clear, written technical feedback.
• Develop guidelines and detailed rubrics/evaluation frameworks for assessing training pipeline design, distributed systems reasoning, and kernel-level optimization across tasks.
• Collaborate with other subject matter experts to ensure consistency and accuracy in training data.

• 2+ years of professional experience in ML infrastructure, MLOps, or ML systems engineering at a recognized, top-tier organization.
• Hands-on production experience with JAX and/or PyTorch at scale.
• Experience in writing or optimizing custom GPU kernels using Pallas (JAX) or Triton.
• Demonstrable career progression.
• Ability to commit to at least 40 hours/week during weekdays.
• Strong written communication skills with the ability to clearly explain complex technical decisions.

This is a W-2 employment position with Cincinnatus LLC, offering a 40-hour full-time engagement with no conflicts or other commitments.

Hourly compensation ranges from $70 to $110.

Applicants must be eligible to work in the United States.