MLOps Engineer (JAX, PyTorch, Pallas/Triton)

Join a leading AI lab’s cutting-edge GenAI team to be at the core of the AI revolution, where your expertise fuels the development of the most advanced Large Language Models.

1. Overview

Join a leading AI lab’s cutting-edge GenAI team and help build foundational AI models from the ground up. We’re seeking talented MLOps Engineers with deep, hands-on expertise in modern ML frameworks — specifically JAX, PyTorch, and kernel-level programming (Pallas/Triton). This role involves AI model training and evaluation work, including writing and assessing MLOps tasks and solutions to generate high-quality training data for frontier AI systems.

This is a W-2 employment position with Cincinnatus LLC, with the opportunity to be placed at a leading AI Lab as part of their extended workforce. This is a 40-hour full-time engagement, with no conflicts/no other engagements.

2. Key Responsibilities

• Guide research and engineering teams to close knowledge gaps and improve AI model performance in MLOps, training infrastructure, and ML framework-level topics.

• Design challenging, domain-relevant tasks, and write accurate and well-structured solutions to MLOps and ML systems problems.

• Evaluate MLOps tasks and solutions and provide clear, written technical feedback.

• Develop guidelines and detailed rubrics/evaluation frameworks to assess 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.

3. Core Qualifications

• 2+ years of dedicated 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 writing or optimizing custom GPU kernels using Pallas (JAX) or Triton.

• Demonstrable career progression.

• Ability to engage reliably for at least 40 hours/week during weekdays.

• Strong written communication skills and the ability to explain complex technical decisions clearly.

About Cincinnatus LLC:
Cincinnatus LLC is an enterprise staffing company that partners with leading technology companies to source and employ highly skilled professionals for contingent and contract-based opportunities. Cincinnatus serves as the employer of record for these engagements, providing W-2 employment, payroll, benefits, and compliance, while placing employees directly within client teams to work on high-impact initiatives.

Equal Employment Opportunity:
Cincinnatus is proud to be an Equal Employment Opportunity employer. We do not discriminate based upon race, religion, color, national origin, sex (including pregnancy, childbirth, reproductive health decisions, or related medical conditions), sexual orientation, gender identity, gender expression, age, status as a protected veteran, status as an individual with a disability, genetic information, political views or activity, or any other legally protected characteristic.