About this role
Build and improve large-scale training pipelines and evaluation workflows for foundational generative AI models, with a focus on ML framework-level engineering and GPU kernel optimization. You will design and author challenging MLOps tasks and solutions, evaluate and provide technical feedback, and create rubrics that ensure high-quality training data for frontier AI systems.
Key Responsibilities- Work with research and engineering teams to close knowledge gaps and improve model performance across MLOps, training infrastructure, and framework-level topics.
- Design domain-relevant, technically challenging tasks and produce accurate, well-structured solutions for MLOps and ML systems problems.
- Evaluate MLOps tasks and submitted solutions, providing clear, written technical feedback.
- Develop guidelines and detailed rubrics to assess training pipeline design, distributed systems reasoning, and kernel-level optimization.
- Collaborate with other subject matter experts to maintain consistency and accuracy in training data and evaluation standards.
- Minimum 2 years of professional experience in ML infrastructure, MLOps, or ML systems engineering at a recognized, top-tier organization.
- Production experience with JAX and/or PyTorch at scale.
- Experience writing or optimizing custom GPU kernels using Pallas (for JAX) or Triton.
- Demonstrable career progression in relevant roles.
- Strong written communication skills, able to explain complex technical decisions clearly.
- Able to engage reliably for at least 40 hours per week on weekdays, with no other engagements or conflicts.
- Employment type: hourly, W-2 employment through Cincinnatus LLC, which acts as the employer of record and places employees within client teams.
- Assignment: placement at a leading AI lab as part of their extended workforce.
- Commitment: full-time, 40 hours per week, weekday availability required, no concurrent engagements permitted.
- Location: United States.
Hourly rate range: 70 to 110 per hour.
EligibilityPosition is offered as W-2 employment through Cincinnatus LLC. Candidates must be able to accept W-2 employment in the United States and meet the stated full-time availability requirement of 40 hours per week on weekdays.
Equal Employment Opportunity: Cincinnatus LLC provides equal employment opportunities and does not discriminate on the basis of legally protected characteristics.