RemoteLeads Logo
Sign Up Sign In

MLOps Engineer (JAX, PyTorch, Pallas/Triton)

6 days, 4 hours 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 75

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

Janea Systems

Senior Software Engineer ML - Contractor position

Janea Systems 11-50 Internet Software & Services

Janea Systems (USA) is seeking a Senior ML Engineer contractor to support remote, client-facing consulting engagements that take enterprise ML and LLM systems from prototype to production.

United States Contract Senior Machine Learning Engineer Software Engineer
Apache Airflow AWS Azure CI/CD Dagster Datadog GCP Kubeflow Kubernetes LLM MLOps NLP Prefect Python PyTorch TensorFlow
3 hours, 23 minutes ago
Verve

Machine Learning Engineer

Verve
Remote
Machine Learning Engineer
AWS Machine Learning MLflow MLOps Python PyTorch Scikit-learn SQL TensorFlow
1 week, 6 days ago
CodeRoad

Machine Learning Operations Engineer (MLOps)

CodeRoad 51-250 Internet Software & Services

Coderoad is hiring a remote MLOps Engineer to design, deploy, and maintain autonomous AI systems for cross-functional projects in the LATAM region.

Latin America Contract Mid Level Machine Learning Engineer
AWS Bash CI/CD Docker GCP Git Grafana Kubernetes Prometheus Python PyTorch Terraform Vertex AI
4 weeks, 2 days ago
Xenon7

Senior Data Scientist - Europe

Xenon7 Internet Software & Services

Xenon7 is hiring a Senior Data Scientist for a remote Europe-based contract to build a document intelligence and discovery platform that turns scanned chemical PDFs into searchable, actionable knowledge bases.

Poland Serbia Croatia Spain France Germany Contract Senior Data Scientist Machine Learning Engineer
AWS Computer Vision Deep Learning Generative AI LLM Machine Learning Python PyTorch Scikit-learn TensorFlow
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