Remote JAX ML Ops Engineer (Contract)

Mercor; New York (NY); Remote; USD 96,432 - 151,536; Full time

Job summary

Mercor is seeking a JAX Expert to guide research teams in improving AI model performance through MLOps. This remote position requires a commitment of 40 hours per week and offers competitive compensation between $70 to $110 per hour.

The ideal candidate will have at least 2 years of experience in ML infrastructure or MLOps, along with hands-on experience using JAX at scale. Strong communication skills and the ability to handle complex technical challenges are essential for this role.

Qualifications

• 2+ years of experience in ML infrastructure, MLOps, or ML systems engineering.
• Hands-on production experience with JAX at scale.
• Experience with custom GPU kernels.

Responsibilities

• Guide research and engineering teams to improve AI model performance in MLOps.
• Design challenging tasks and write solutions to ML systems problems.
• Evaluate MLOps tasks and provide technical feedback.

Skills

MLOps

ML infrastructure

JAX

GPU kernels

Tools

Pallas

Triton

Job description

Mercor is seeking a JAX Expert to guide research teams in improving AI model performance through MLOps. This remote position requires a commitment of 40 hours per week and offers competitive compensation between $70 to $110 per hour.

The ideal candidate will have at least 2 years of experience in ML infrastructure or MLOps, along with hands-on experience using JAX at scale. Strong communication skills and the ability to handle complex technical challenges are essential for this role.