Remote MLOps Engineer (JAX at Scale)

NLP PEOPLE; San Francisco (CA); On-site; USD 96,432 - 151,536; Full time

Job summary

NLP PEOPLE is seeking a JAX Expert for a remote contract role based in San Francisco, CA. The position involves guiding research teams in improving AI models and designing domain-relevant tasks while ensuring technical accuracy.

The ideal candidate has over 2 years of experience in MLOps or ML systems engineering and hands-on production experience with JAX. Strong communication skills and the ability to commit to 40 hours per week are essential.

Qualifications

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

Responsibilities

• Guide research teams to improve AI model performance.
• Design domain-relevant tasks and write structured solutions.
• Evaluate MLOps tasks and provide technical feedback.
• Develop guidelines for assessing training pipeline design.
• Collaborate with experts for consistency in training data.

Skills

ML infrastructure

MLOps

ML systems engineering

JAX

GPU kernels optimization

Written communication

Job description

NLP PEOPLE is seeking a JAX Expert for a remote contract role based in San Francisco, CA. The position involves guiding research teams in improving AI models and designing domain-relevant tasks while ensuring technical accuracy.

The ideal candidate has over 2 years of experience in MLOps or ML systems engineering and hands-on production experience with JAX. Strong communication skills and the ability to commit to 40 hours per week are essential.