Remote MLOps Engineer; JAX, PyTorch, Pallas​/Triton — IN - AI Trainer

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 is a W‑2 employment position with Cincinnatus LLC, requiring a commitment of 40 hours per week (during weekdays). This position will be placed at a leading AI Lab as part of their extended workforce.

• 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.

• 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 30 hours per week during weekdays.
• Strong written communication skills and the ability to explain complex technical decisions clearly.

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.