MLOps Engineer (JAX, PyTorch, Pallas/Triton) for Large Language Model Training
Design challenging tasks, write reference solutions, and build rubrics for large language model training. Collaborate with ML subject matter experts and maintain consistency across training data. 2+ years of professional experience in ML Infrastructure, MLOps, or ML Systems Engineering required.
Key Highlights
Key Responsibilities
Technical Skills Required
Benefits & Perks
Job Description
Hiring MLOps Engineers (JAX, PyTorch, Pallas/Triton) to help train the next generation of large language models at a leading frontier AI lab. Remote from India.
$35–$45/hour USD, W-2 contract, 30–40 hrs/week weekdays.
Para AI Labs is sourcing this role on behalf of Mercor — a well-established AI hiring platform — for a leading frontier AI lab's GenAI team. You won't be labeling data. You'll be designing the technical problems, writing reference solutions, and building the rubrics that feed directly into how the next generation of LLMs reason about MLOps and ML systems.
The essentials
- Pay: $35–$45/hour USD (strong rates for India-based MLOps work)
- Work mode: Fully remote from anywhere in India
- Commitment: 30–40 hrs/week, weekdays only
- Employment: W-2 via Cincinnatus LLC (the employer of record)
- Placement: A leading frontier AI lab's GenAI team
What you'll do
- Design challenging tasks across MLOps, ML infrastructure, and ML systems for use in model training
- Write accurate, well-structured reference solutions covering distributed training, training pipelines, and GPU kernel optimization
- Evaluate submissions from other engineers and provide clear, written technical feedback
- Develop rubrics and guidelines for assessing training pipeline design, distributed systems reasoning, and kernel-level optimization
- Collaborate with other ML subject matter experts to maintain consistency and accuracy across the training data
- Guide research and engineering teams toward closing specific knowledge gaps in ML framework internals
Interested in remote work opportunities in Machine Learning & AI? Discover Machine Learning & AI Remote Jobs featuring exclusive positions from top companies that offer flexible work arrangements.
What you need
- 2+ years of 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 — real training workloads, not coursework or hobby projects
- Experience writing or optimizing custom GPU kernels with Pallas (JAX) or Triton
- Demonstrable career progression
- Strong written English — you can explain complex technical decisions clearly
- Reliable availability for at least 30 hours/week on weekdays
Browse our curated collection of remote jobs across all categories and industries, featuring positions from top companies worldwide.
Why this is worth your time
- Top-tier USD rates for remote India-based MLOps work
- Direct, measurable impact on a frontier AI lab's next-generation models
- Proper W-2 employment via Cincinnatus — no invoice chasing, no contractor tax surprises, no weekend work
- Work alongside senior ML engineers and researchers from across the global AI ecosystem
About Para AI Labs
Para AI Labs is the curated destination for high-quality AI training and evaluation roles from leading platforms including Mercor, Scale AI, Turing, and others. We help skilled engineers and domain experts find paid work on frontier AI projects without searching across ten different platforms.
How to apply
This role is placed through Mercor, the AI hiring platform handling the application process. Apply directly here:
👉 https://t.mercor.com/qMcxT
Questions? Email hello@parailabs.in or browse our other open AI roles at parailabs.in/projects.
Similar Jobs
Explore other opportunities that match your interests
Mercor
Stealth Startup
