MLOps Engineer, LLM Performance (Kernels, Profiling, Serving)

Mercor

India

Remote

INR 11,642,000 - 15,522,000

Part time

2 days ago
Be an early applicant

Get a reply from this employer

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

Job summary

Mercor is seeking an MLOps Engineer for GenAI teams focused on kernels, profiling, and high-throughput inference serving. The role leverages JAX/XLA, CUDA and modern accelerators, with remote, flexible engagement and weekly payments.

You will contribute to frontier AI model performance and training data quality through robust ML systems tooling. This fully remote, hourly contract role requires 2+ years in ML systems and hands-on experience with accelerator performance engineering, JAX or

Qualifications

  • 2+ years in ML systems, ML infrastructure, model serving, or accelerator performance engineering.
  • Experience with designing and debugging high-throughput inference workloads and accelerator kernels.
  • Strong knowledge of modern ML frameworks and compilers (JAX, XLA, PyTorch) is required.

Responsibilities

  • Design tasks across kernels, profiling, debugging and inference serving, including frontier model scenarios.
  • Guide research teams to improve AI model performance on ML systems, compilers and training infra.
  • Evaluate ML system tasks and provide clear technical feedback.

Skills

MLOps
LLM serving
Kernel programming
Performance profiling
Distributed systems
JAX
PyTorch
CUDA

Tools

JAX/XLA profiler
Kineto
torch.profiler
Nsight
Megatron

Job description

MLOps Engineer, LLM Performance (Kernels, Profiling, Serving)Company: MercorCompensation: $90-$120 per hourCommitment: Hourly contractLocation: Remote

Join a leading AI lab's cutting-edge GenAI team to be at the core of the AI revolution, where your expertise fuels the development of the most advanced Large Language Models.

1. Overview

Join a leading AI lab's cutting-edge GenAI team and help build foundational AI models from the ground up. We're seeking MLOps and ML systems engineers with hands-on experience in large language model infrastructure across any of four areas: accelerator kernel programming, performance profiling and trace analysis, debugging accelerated and distributed workloads, and high-throughput inference serving. Experience with JAX, XLA or TPUs is especially valuable. This role involves AI model training and evaluation work, including writing and assessing ML systems tasks and solutions to generate high-quality training data for frontier AI systems.

This is a fully remote role open to candidates worldwide. Depending on where you are based, your engagement will be arranged through Cincinnatus LLC or one of its trusted employer-of-record partners in your country, with the opportunity to be placed at a leading AI lab as part of their extended workforce. This is a 40-hour full-time engagement, with no conflicts/no other engagements.

2. Key Responsibilities
  • Design challenging tasks across kernels, profiling, debugging and inference serving, including adversarial problems that expose where frontier models fail on accelerator code (for example Pallas kernels on TPU), and write accurate, well-structured solutions to them.
  • Guide research and engineering teams to close knowledge gaps and improve AI model performance on ML systems, compilers and training infrastructure topics.
  • Evaluate ML systems tasks and solutions, and provide clear written technical feedback that stands up to expert review.
  • Develop guidelines and detailed rubrics covering kernel optimization, profiler output interpretation, distributed systems reasoning, and serving throughput and latency trade-offs.
  • Collaborate with other subject matter experts to keep training data consistent and accurate.
3. Core Qualifications
  • 2+ years of hands-on professional experience in ML systems, ML infrastructure, model serving, or accelerator performance engineering. This is a hands-on systems role rather than an applied modelling or data science one.
  • Practical experience in at least one of the following, with more than one a strong plus: writing or optimizing custom kernels (Pallas, Triton, CUDA, CUTLASS); performance profiling and trace analysis (JAX/XLA profiler, Kineto, torch.profiler, Nsight); debugging distributed or accelerator-bound workloads; serving large language models at scale (vLLM, SGLang, TensorRT-LLM, JetStream, KV cache, paged attention, continuous batching).
  • Working production experience with JAX and/or PyTorch. Framework-level depth is a strong plus: custom operators, sharding and distributed training (pjit/shard_map, FSDP, Megatron, MaxText), or compiler and graph-level work (XLA/StableHLO, torch.compile, MLIR).
  • Experience with modern accelerators such as TPU (v4 to v6e), H100, B200, Trainium or MI300, and the ability to reason about throughput, latency and memory trade-offs.
  • Demonstrable career progression.
  • Ability to engage reliably for at least 40 hours/week during weekdays.
  • Strong written communication skills and the ability to explain complex technical decisions clearly.
About Cincinnatus LLC:

Cincinnatus LLC is an enterprise staffing company that partners with leading technology companies to source and engage highly skilled professionals for contingent and contract-based opportunities. Working directly or through local employer-of-record partners, Cincinnatus handles the employment or contracting relationship, payroll, benefits where applicable, and compliance in each country, while placing professionals directly within client teams to work on high-impact initiatives.

Equal Employment Opportunity:

Cincinnatus is proud to be an Equal Employment Opportunity employer. We do not discriminate based upon race, religion, color, national origin, sex (including pregnancy, childbirth, reproductive health decisions, or related medical conditions), sexual orientation, gender identity, gender expression, age, status as a protected veteran, status as an individual with a disability, genetic information, political views or activity, or any other legally protected characteristic. We consider all qualified applicants without regard to legally protected characteristics and provide reasonable accommodations upon request.

Contract and Payment Terms

You will be engaged as an independent contractor. This is a fully remote role that can be completed on your own schedule. Projects can be extended, shortened, or concluded early depending on needs and performance. Your work at Mercor will not involve access to confidential or proprietary information from any employer, client, or institution. Payments are weekly on Stripe or Wise based on services rendered.

Please note: We are unable to support H1-B or STEM OPT candidates at this time.

About Mercor

Mercor partners with leading AI labs and enterprises to train frontier models using human expertise. You will work on projects that focus on training and enhancing AI systems. You will be paid competitively, collaborate with leading researchers, and help shape the next generation of AI systems in your area of expertise.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Machine Learning Expert
Machine Learning Expert

Mercor • India

Remote
INR 9,055,000 - 15,522,000
Remote work
Weekly payments (Stripe/Wise)
Software Engineer (AI-Native Platform & Integrations)
Software Engineer (AI-Native Platform & Integrations)

Mercor • India

Remote
INR 8,451,000 - 14,085,000
Electrical & Hardware Engineering Experts
Electrical & Hardware Engineering Experts

Mercor • India

Remote
INR 1,200,000 - 2,400,000
Legacy Codebase Migration Expert
Legacy Codebase Migration Expert

Mercor • India

Remote
INR 11,268,000 - 16,901,000
Weekly payments via Stripe or Wise
Fully remote role
Flexible schedule
Business Operations Expert
Business Operations Expert

Mercor • India

Remote
INR 11,642,000 - 15,522,000
Software & Firmware Engineering Experts
Software & Firmware Engineering Experts

Mercor • India

Remote
INR 5,127,000 - 11,962,000
Weekly payments via Stripe/Wise
Machine Learning Engineer Talent Network
Machine Learning Engineer Talent Network

Mercor • India

Remote
INR 6,408,000 - 11,535,000
Life, Physical & Social Science Expert
Life, Physical & Social Science Expert

Mercor • India

Remote
INR 7,761,000 - 15,522,000
English Linguistic Experts
English Linguistic Experts

Mercor • India

Remote
INR 2,563,000 - 4,272,000
Film & Video Craft Expert (Camera, VFX, Editing & Performance)
Film & Video Craft Expert (Camera, VFX, Editing & Performance)

Mercor • India

Remote
INR 9,055,000 - 12,935,000
Referral reward up to $1,600