← All jobs in United States

Remote

MLOps Engineer - JAX - PyTorch - Pallas/Triton

Weekday AI · United States · United States

Remote Part time USD 70–110 / hour Posted Open until 18 Oct 2026
  • Remote Enjoy a fully remote engagement with meaningful technical challenges. — job description

The job, in the employer's words

This role is for one of our clients

Compensation: $70-$110 per hour

Join a cutting-edge AI research initiative at the forefront of Generative AI and contribute to the development of next-generation Large Language Models. We are seeking experienced MLOps Engineers with deep expertise in modern machine learning frameworks, large-scale training infrastructure, and kernel-level optimization.

In this role, you'll leverage your knowledge of JAX, PyTorch, and custom GPU kernel programming (Pallas/Triton) to create, evaluate, and refine high-quality technical tasks that help train frontier AI systems. You'll collaborate with AI researchers and engineering teams to improve model reasoning across MLOps, distributed training, and ML infrastructure topics.

This is a full-time, 40-hour-per-week remote engagement requiring full weekday availability.

Requirements

Key Responsibilities

  • Partner with research and engineering teams to strengthen AI model capabilities in MLOps, ML infrastructure, and large-scale training systems.
  • Design challenging, real-world MLOps and machine learning systems tasks that reflect production engineering scenarios.
  • Develop accurate, well-documented solutions to complex ML infrastructure and training pipeline problems.
  • Review and evaluate technical tasks and AI-generated solutions, providing clear and actionable written feedback.
  • Create detailed evaluation rubrics and scoring frameworks for topics including:
    • Distributed training architectures
    • ML pipeline design
    • Infrastructure optimization
    • Kernel-level programming
    • Performance tuning

  • Collaborate with fellow subject matter experts to maintain consistency, quality, and technical accuracy across training datasets.
  • Contribute domain expertise to improve the reasoning capabilities of advanced AI systems.

Required Qualifications

  • Minimum 2 years of professional experience in MLOps, Machine Learning Infrastructure, or ML Systems Engineering within a recognized technology organization.
  • Hands-on production experience with JAX and/or PyTorch in large-scale machine learning environments.
  • Practical experience developing or optimizing custom GPU kernels using Pallas (JAX) or Triton.
  • Strong understanding of distributed training systems, model optimization, and scalable ML infrastructure.
  • Demonstrated career growth and increasing technical responsibility.
  • Availability to work 40 hours per week during standard weekday business hours.
  • Excellent written communication skills with the ability to clearly explain technical concepts and architectural decisions.

Preferred Skills

  • Experience designing and optimizing large-scale ML training pipelines.
  • Knowledge of distributed computing and GPU performance optimization.
  • Familiarity with evaluation methodologies for AI models and ML systems.
  • Experience collaborating with research teams on advanced machine learning projects.
  • Passion for advancing AI infrastructure and frontier model development.

Why Join

  • Help build and improve next-generation Large Language Models.
  • Work alongside leading AI researchers and experienced machine learning engineers.
  • Apply your expertise to high-impact projects involving large-scale ML systems and infrastructure.
  • Contribute directly to the development of cutting-edge AI technologies.
  • Enjoy a fully remote engagement with meaningful technical challenges.

Equal Opportunity

We welcome applications from qualified professionals regardless of legally protected characteristics and are committed to providing reasonable accommodations throughout the application and engagement process upon request.

Contract & Payment Terms

  • Engagement is offered on an independent contractor basis.
  • This is a fully remote opportunity that can be completed according to your own schedule.
  • Project duration may be extended, shortened, or concluded based on project requirements and individual performance.
  • The engagement does not require access to confidential or proprietary information belonging to any current employer, client, or institution.
  • Payments are processed weekly through Stripe or Wise based on approved work completed.
  • Please note: Applicants requiring H-1B sponsorship or participating in the STEM OPT program are not eligible for this opportunity.

Benefits mentioned

  • Join a cutting-edge AI research initiative at the forefront of Generative AI and contribute to the development of next-generation Large Language Models. We are seeking experienced MLOps Engineers with deep expertise in modern machine learning frameworks, large-scale training infrastructure, and kernel-level optimization.
  • In this role, you'll leverage your knowledge of JAX, PyTorch, and custom GPU kernel programming (Pallas/Triton) to create, evaluate, and refine high-quality technical tasks that help train frontier AI systems. You'll collaborate with AI researchers and engineering teams to improve model reasoning across MLOps, distributed training, and ML infrastructure topics.
  • This is a full-time, 40-hour-per-week remote engagement requiring full weekday availability.
  • Help build and improve next-generation Large Language Models.
  • Work alongside leading AI researchers and experienced machine learning engineers.
  • Apply your expertise to high-impact projects involving large-scale ML systems and infrastructure.
  • Contribute directly to the development of cutting-edge AI technologies.
  • Enjoy a fully remote engagement with meaningful technical challenges.

Requirements mentioned

  • Minimum 2 years of professional experience in MLOps, Machine Learning Infrastructure, or ML Systems Engineering within a recognized technology organization.
  • Hands-on production experience with JAX and/or PyTorch in large-scale machine learning environments.
  • Practical experience developing or optimizing custom GPU kernels using Pallas (JAX) or Triton.
  • Strong understanding of distributed training systems, model optimization, and scalable ML infrastructure.
  • Demonstrated career growth and increasing technical responsibility.
  • Availability to work 40 hours per week during standard weekday business hours.
  • Excellent written communication skills with the ability to clearly explain technical concepts and architectural decisions.
  • Experience designing and optimizing large-scale ML training pipelines.
  • Knowledge of distributed computing and GPU performance optimization.
  • Familiarity with evaluation methodologies for AI models and ML systems.
  • Experience collaborating with research teams on advanced machine learning projects.
  • Passion for advancing AI infrastructure and frontier model development.

No Mondays shows this listing as the employer published it. Check the details on the employer's site before you apply.

More remote jobs in United States

Natera · US Remote · today

Medical Science Liaison - Rare Disease

POSITION SUMMARY: The Medical Science Liaison (MSL), Rare Disease is a field-based scientific expert and a key scientific resource for clinicians, academic institutions, …

RemoteUSD 131k–163k / year
Vercel · San Francisco · today

GTM Engineer

About Vercel: Vercel is the agentic infrastructure company, freeing people and agents to ship what's next. For more than a decade we've helped builders move from idea to …

RemoteUSD 180k–295k / year
Cortica - Neurodevelopmental · San Diego · today

Enrollment Specialist - Remote

Cortica is looking for an Enrollment Specialist to join our growing multi-disciplinary team! Under the general direction of the Cortica Head of Client Experience, the Enr…

RemoteUSD 23–25 / hour
Sayari · Remote - US · today

Field Intelligence Analyst - US

About Sayari: Sayari is the judgment infrastructure for trustworthy AI in economic security and commercial risk. The Sayari Commercial World Model resolves 11.7B+ primary…

Remote
Symmetry AI · Denver · today

Forward Deployed Solution Engineer

Symmetry is the payroll infrastructure for software & payroll platforms powering the paychecks of over 64 million workers each year. Our fully integrated suite of payroll…

Remote
Wiz, Inc. · Remote - USA · today

Head of Strategic Programs Operations

Come join the organization that is redefining security for the AI era. As one of the fastest-growing startups ever, we enable teams to secure cloud and AI applications by…

RemoteUSD 234k–300k / year

23,426 live roles · See all →

Other roles at Weekday AI

Weekday AI · India · yesterday

Windows Administrator

𝗧𝗵𝗶𝘀 𝗿𝗼𝗹𝗲 𝗶𝘀 𝗳𝗼𝗿 𝗼𝗻𝗲 𝗼𝗳 𝘁𝗵𝗲 𝗪𝗲𝗲𝗸𝗱𝗮𝘆'𝘀 𝗰𝗹𝗶𝗲𝗻𝘁𝘀 𝗦𝗮𝗹𝗮𝗿𝘆 𝗿𝗮𝗻𝗴𝗲: 𝗥𝘀 𝟱𝟬𝟬𝟬𝟬𝟬 - 𝗥𝘀 𝟭𝟱𝟬𝟬𝟬𝟬𝟬 (𝗶𝗲 𝗜𝗡𝗥 𝟱-𝟭𝟱 𝗟𝗣𝗔) Experience: 6+ yrs Location: Remote (India) We are looking for an exp…

Remote
Weekday AI · India · yesterday

QA Engineer

𝗧𝗵𝗶𝘀 𝗿𝗼𝗹𝗲 𝗶𝘀 𝗳𝗼𝗿 𝗼𝗻𝗲 𝗼𝗳 𝘁𝗵𝗲 𝗪𝗲𝗲𝗸𝗱𝗮𝘆'𝘀 𝗰𝗹𝗶𝗲𝗻𝘁𝘀 𝗦𝗮𝗹𝗮𝗿𝘆 𝗿𝗮𝗻𝗴𝗲: 𝗥𝘀 𝟭𝟬𝟬𝟬𝟬𝟬𝟬 - 𝗥𝘀 𝟮𝟱𝟬𝟬𝟬𝟬𝟬 (𝗶𝗲 𝗜𝗡𝗥 𝟭𝟬-𝟮𝟱 𝗟𝗣𝗔) Experience: 6+ yrs Location: Remote (India) We are looking for an e…

Remote
Weekday AI · India · yesterday

Accounts Payable Specialist

𝗧𝗵𝗶𝘀 𝗿𝗼𝗹𝗲 𝗶𝘀 𝗳𝗼𝗿 𝗼𝗻𝗲 𝗼𝗳 𝘁𝗵𝗲 𝗪𝗲𝗲𝗸𝗱𝗮𝘆'𝘀 𝗰𝗹𝗶𝗲𝗻𝘁𝘀 𝗦𝗮𝗹𝗮𝗿𝘆 𝗿𝗮𝗻𝗴𝗲: 𝗥𝘀 𝟭𝟯𝟬𝟬𝟬𝟬𝟬 - 𝗥𝘀 𝟭𝟲𝟬𝟬𝟬𝟬𝟬 (𝗶𝗲 𝗜𝗡𝗥 𝟭𝟯-𝟭𝟲 𝗟𝗣𝗔) Experience: 2+ yrs Location: Remote (India), India We are looking f…

Remote
Weekday AI · India · 2 days ago

SF OMS Developer

𝗧𝗵𝗶𝘀 𝗿𝗼𝗹𝗲 𝗶𝘀 𝗳𝗼𝗿 𝗼𝗻𝗲 𝗼𝗳 𝘁𝗵𝗲 𝗪𝗲𝗲𝗸𝗱𝗮𝘆'𝘀 𝗰𝗹𝗶𝗲𝗻𝘁𝘀 𝗦𝗮𝗹𝗮𝗿𝘆 𝗿𝗮𝗻𝗴𝗲: 𝗥𝘀 𝟭𝟬𝟬𝟬𝟬𝟬𝟬 - 𝗥𝘀 𝟮𝟱𝟬𝟬𝟬𝟬𝟬 (𝗶𝗲 𝗜𝗡𝗥 𝟭𝟬-𝟮𝟱 𝗟𝗣𝗔) Experience: 4+ yrs Location: Remote (India) Job Type: Remote We are…

Remote
Weekday AI · India · 2 days ago

gtm engineer

𝗧𝗵𝗶𝘀 𝗿𝗼𝗹𝗲 𝗶𝘀 𝗳𝗼𝗿 𝗼𝗻𝗲 𝗼𝗳 𝘁𝗵𝗲 𝗪𝗲𝗲𝗸𝗱𝗮𝘆'𝘀 𝗰𝗹𝗶𝗲𝗻𝘁𝘀 𝗦𝗮𝗹𝗮𝗿𝘆 𝗿𝗮𝗻𝗴𝗲: 𝗥𝘀 𝟭𝟴𝟬𝟬𝟬𝟬𝟬 - 𝗥𝘀 𝟮𝟴𝟬𝟬𝟬𝟬𝟬 (𝗶𝗲 𝗜𝗡𝗥 𝟭𝟴-𝟮𝟴 𝗟𝗣𝗔) Experience: 5+ yrs Location: Remote (India), India Job Type: Full-t…

Remote
Weekday AI · India · 3 days ago

Head Of Business Development

This role is for one of Weekday’s clients Salary range: Rs 2000000 - Rs 2500000 (ie INR 20 - 25 LPA) Min Experience: 6+ years Location: India JobType: full-time We are lo…

RemoteINR 2000k–2500k / year

260 live roles · See all →