Seeking experienced MLOps and ML systems engineers with PyTorch expertise to develop, evaluate, and improve AI training systems and large language models.
JAX Expert
United States
$145,000 – $228,000/yr
June 22, 2026
Job description
Join a leading AI lab's cutting-edge Generative AI team and help build foundational AI models from the ground up.
This opportunity is seeking experienced MLOps Engineers with deep expertise in JAX and kernel-level programming using Pallas or Triton.
The role focuses on AI model training and evaluation work, including designing MLOps tasks, evaluating technical solutions, and creating high-quality training data for frontier AI systems.
This is a full-time W-2 employment opportunity through Cincinnatus LLC, with placement on a leading AI lab team as part of its extended workforce.
Key Responsibilities
AI Training & Evaluation
- Design challenging, domain-relevant MLOps and ML systems tasks
- Create accurate and well-structured technical solutions
- Evaluate task submissions and provide written technical feedback
- Help improve AI model performance through domain expertise
MLOps & Infrastructure
Guide research and engineering teams on:
- MLOps practices
- ML infrastructure
- Training systems
- Distributed systems
- Framework-level engineering topics
Develop evaluation frameworks and grading rubrics for:
- Training pipeline design
- Distributed systems reasoning
- Kernel optimization
- ML systems engineering
Collaboration
- Work with subject matter experts to ensure consistency and accuracy across training datasets
- Help identify knowledge gaps and improve model performance
Core Qualifications
2+ years of professional experience in:
- ML Infrastructure
- MLOps
- ML Systems Engineering
Experience at recognized, top-tier organizations
Hands-on production experience with JAX at scale
Experience building or optimizing custom GPU kernels using:
- Pallas
- Triton
Demonstrated career progression
Ability to work reliably for 40 hours per week
Strong written communication and technical explanation skills
Employment Details
- W-2 employment through Cincinnatus LLC
- Full-time engagement
- 40 hours per week
- Weekday availability required
- No conflicting engagements permitted
Compensation
- $70–110/hour
About Cincinnatus LLC
Cincinnatus LLC is an enterprise staffing company that partners with leading technology organizations to source and employ highly skilled professionals for contract and contingent opportunities.
The company serves as the employer of record for these engagements, providing:
- W-2 employment
- Payroll services
- Benefits administration
- Compliance support
Professionals are placed directly within client teams to contribute to high-impact initiatives.
Equal Opportunity
Cincinnatus LLC is an Equal Employment Opportunity employer and considers qualified applicants regardless of legally protected characteristics.
Reasonable accommodations are available upon request.
Apply now
You will be redirected to the company's website to complete your application.
SWE - Machine Learning (Contract) Labelbox • Remote (United States preferred) Shape the data that powers frontier AI Quick facts - Engagement - Hourly, at‑will contractor- Schedule - Fully remote & asynchronous (min.
Innodata (Nasdaq: INOD) is a global data engineering company. We believe that data and Artificial Intelligence (AI) are inextricably linked.
Innodata (Nasdaq: INOD) is a global data engineering company. We believe that data and Artificial Intelligence (AI) are inextricably linked.