AI Infrastructure Specialist - Fully Remote | Upto $130/hr

Role Overview

Mercor is seeking an AI Infrastructure Specialist / MLOps Engineer (JAX, PyTorch, Pallas/Triton) for a fully remote contract role. You will help research and engineering teams improve AI model performance by closing knowledge gaps across MLOps, training infrastructure, and ML framework-level topics.

Responsibilities

• Guide research and engineering teams to close knowledge gaps and improve AI model performance in MLOps/training infrastructure/ML framework topics
• Design challenging, domain-relevant tasks across multiple specializations
• Write accurate, well-structured solutions to MLOps and ML systems problems
• Evaluate MLOps tasks and solutions
• Provide clear, written technical feedback
• Create guidelines and detailed rubrics/evaluation frameworks for:
  • training pipeline design
  • distributed systems reasoning
  • kernel-level optimization
• Collaborate with other subject matter experts to ensure consistency and accuracy in training data

Requirements

• 5+ years of professional experience in ML infrastructure, MLOps, or ML systems engineering at a top-tier organization
• Hands-on production experience with JAX and/or PyTorch at scale, including:
  • distributed training strategies (FSDP, tensor parallelism, pipeline parallelism)
  • memory optimization
  • framework-level debugging
• Experience writing or optimizing custom GPU kernels using:
  • Pallas (JAX) or Triton
  • tiling strategies, memory layout design, and kernel fusion
• Demonstrable career progression
• Ability to engage reliably for 30+ hours/week on weekdays
• Strong written communication skills to explain complex technical decisions

Contract & Logistics

• Fully remote
• W-2 employment via Cincinnatus LLC
• Compensation: $90–$130/hour