RemoteLeads Logo
Sign Up Sign In

MLOps Engineer (JAX, PyTorch, Pallas/Triton)

2 months, 1 week ago
United States
Contract
Junior
Machine Learning Engineer
Software Development
MLOps PyTorch
Apply Now
Weekday

Weekday

Weekday helps companies hire engineers who are vouched by other software engineers, enabling passive income for engineers. They offer services like drafting outreach messages, shortlisting candidates, and conducting reference checks. Backed by Y Combin...

Construction & Engineering
11-50
Founded 2020
View All Jobs 126

Description

  • Partner with research and engineering teams to improve 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 evaluation rubrics and scoring frameworks for distributed training, ML pipeline design, infrastructure optimization, kernel-level programming, and performance tuning.
  • Collaborate with 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.

Requirements

  • 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 (preferred).
  • Knowledge of distributed computing and GPU performance optimization (preferred).
  • Familiarity with evaluation methodologies for AI models and ML systems (preferred).
  • Experience collaborating with research teams on advanced machine learning projects (preferred).
  • Passion for advancing AI infrastructure and frontier model development (preferred).
  • H-1B sponsorship and STEM OPT candidates are not eligible for this opportunity.

Benefits

  • Compensation of $70-$110 per hour.
  • Fully remote engagement.
  • Independent contractor arrangement.
  • Weekly payments via Stripe or Wise.
  • Potential for project extension based on requirements and performance.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

AiFi

Sr. AI Systems Engineer

AiFi 51-250 Internet Software & Services

AiFi is hiring a computer vision engineering leader to guide research and development for scene understanding systems that calibrate large camera networks and interpret physical spaces across deployed customer environments.

Portugal Spain Contract Senior Machine Learning Engineer
Azure Computer Vision Docker Go gRPC Kubernetes MLOps OpenAPI Python REST API
1 day, 18 hours ago
Zone & Co

AI Software Engineer

Zone & Co 251-1K Diversified Financial Services

Zone & Co is seeking an AI Software Engineer to design, deploy, and optimize production-ready AI systems embedded in its Oracle NetSuite financial operations platform.

Spain Contract Junior Machine Learning Engineer
AWS Azure JavaScript Machine Learning Python SQL
1 week, 3 days ago
Zone & Co

AI Software Engineer

Zone & Co 251-1K Diversified Financial Services

Zone & Co is seeking an AI Software Engineer to build and deploy production-ready AI solutions that enhance its Oracle NetSuite financial operations platform.

Czech Republic Contract Junior Machine Learning Engineer
AWS Azure JavaScript Machine Learning Python SQL
1 week, 4 days ago
Aspenview Technology Partners

Mid-Level AI Platform Engineer — SAIQ

Aspenview Technology Partners Internet Software & Services

AspenView Technology Partners is seeking a Mid-Level AI Platform Engineer for a fully remote contract supporting an AI-native SaaS platform, owning production operations, quality, releases, and incremental product delivery.

United States Contract Mid Level Machine Learning Engineer
Git Salesforce SAP
1 month ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers