MLOps Engineer (JAX, PyTorch, Pallas/Triton) | $35–45/hr | Remote (US) | W-2 Contract, 40 hrs/week
Join a leading AI lab's GenAI team and play a central role in building foundational Large Language Models from the ground up. This is a high-impact W-2 contract position (via Cincinnatus LLC) embedded within a top-tier AI lab's extended workforce, focused on ML infrastructure, training pipelines, and kernel-level optimization.
Key Responsibilities
- Guide research and engineering teams to close knowledge gaps and improve AI model performance across MLOps, training infrastructure, and ML framework topics.
- Design challenging, domain-relevant tasks and write accurate, well-structured solutions to MLOps and ML systems problems.
- Evaluate MLOps tasks and solutions, providing clear and detailed written technical feedback.
- Develop guidelines, rubrics, and evaluation frameworks to assess training pipeline design, distributed systems reasoning, and kernel-level optimization.
- Collaborate with subject matter experts to ensure consistency and accuracy in training data.
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 in ML engineering or related fields.
- Availability for at least 30–40 hours per week during weekdays.
- Strong written communication skills with the ability to explain complex technical decisions clearly.
About the Employer
Cincinnatus LLC is an enterprise staffing company that partners with leading technology organizations to place highly skilled professionals in contract roles. Cincinnatus serves as the employer of record, providing W-2 employment, payroll, benefits, and compliance support while embedding employees within client teams on high-impact initiatives.
Cincinnatus LLC is an Equal Employment Opportunity employer and does not discriminate on the basis of race, religion, gender, age, disability, veteran status, or any other protected characteristic.
