Machine Learning Engineer - ML Training Platform

Pluralis Research

California

Remote

USD 120,000 - 170,000

Full time

9 days ago

Get a reply from this employer

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

Benefits offered by this job

Equity-heavy package
Remote-first culture
Visa sponsorship and relocation

Job summary

Pluralis Research is seeking an experienced infrastructure engineer to design and scale the platform for Protocol Learning across consumer-grade hardware and cloud resources.

You will manage multi-cloud deployments (AWS/GCP/Azure), distributed training pipelines, and real-time networking, ensuring fault tolerance and high availability for ongoing experimentation.

The role emphasizes remote-first collaboration, strong SRE practices, and ownership of the stack from orchestration to observability.

Qualifications

  • Experience designing multi-cloud infrastructure and large-scale compute pipelines.
  • Strong background in distributed ML workflows and checkpointing.
  • Proficient with Python, async programming, and reliability tooling.

Responsibilities

  • Architect fault-tolerant multi-cloud infrastructure for distributed ML systems.
  • Build and scale GPU workloads, data streaming, and model versioning pipelines.
  • Develop observability, monitoring, and incident response practices.

Skills

Infrastructure engineering
Distributed systems
Python engineering
Observability
SRE practices
Startup experience

Education

Bachelor's degree in Computer Science or similar

Tools

Pulumi
Terraform
Docker
Kubernetes
NVIDIA runtime
Prometheus/Grafana
Cloud SDKs

Job description

Pluralis Research works on Protocol Learning: training and serving large models in a fully decentralized way on small consumer-grade devices connected via the internet. Despite being dismissed as infeasible, we have made significant advances on this problem, most recently Agora, a permissionless run that pretrained an 8B model from scratch on consumer GPUs spread over the internet, with no single participant ever holding the full weights (tech report). While many of the core research problems have been solved, Protocol Learning unlocks a series of new challenges. For the mission in full, read A Third Path: Protocol Learning.

Our training and inference doesn't happen in a datacenter. It happens on consumer nodes and cloud instances that are not co-located, connected by ordinary internet, joining and leaving mid-run. Your primary role is to architect, build, and scale the platform that keeps continuous experimentation and large-scale training running on top of that: infrastructure orchestration, distributed compute, and the services that tie them together.

Key Responsibilities
  • Multi-cloud infrastructure: Design the resource management systems that provision and orchestrate compute across AWS, GCP, and Azure with infrastructure-as-code (Pulumi/Terraform). Handle dynamic scaling, state synchronization, and concurrent operations across hundreds of heterogeneous nodes.
  • Distributed training and inference systems: Architect fault-tolerant infrastructure for distributed ML. GPU clusters, NVIDIA runtime, S3 checkpointing, large-dataset management and streaming, health monitoring, and resilient retry strategies.
  • Real-world networking: Build the systems that simulate and handle real network conditions such as bandwidth shaping, latency injection, packet loss. Managing node churn and keeping data flowing across workers with heterogeneous connectivity.
What We're Looking For
  • Infrastructure and platform engineering (required): Production experience with infrastructure-as-code (Pulumi/Terraform/CloudFormation) managing multi-cloud deployments, Docker/Kubernetes (EKS), GPU workloads, and heterogeneous clusters at scale.
  • Distributed systems and ML infrastructure: You understand distributed training workflows: checkpointing, data sharding, model versioning, long-running job orchestration.
  • Decentralized networking: P2P, NAT traversal, traffic shaping, real bandwidth constraints.
  • Systems programming and reliability: Strong Python engineering (asyncio, concurrency, retry logic, cloud SDKs, CLI tooling) with hands-on observability and SRE practice; Prometheus/Grafana, performance profiling, incident response.
  • Environment fit: You've done this in a startup with heavy service orchestration, or at big-tech scale, and you can show which systems you owned.
  • Mission alignment: You believe Protocol Learning is the viable third path for collective, trustless, and sovereign AI.
Nice to Have
  • Experience with foundation model pre-training, post-training, or RL.
  • Experience at proprietary, open-weight and open-source AI labs
Compensation & Benefits
  • Equity-Heavy Package: We offer significant ownership for key technical contributors in addition to a high base salary.
  • Remote-First Culture: Flexible work environment with team members distributed globally.
  • Visa Sponsorship: Optional full visa sponsorship and relocation support to either Australia or the US.
  • Open Problems: Training and serving frontier models on hardware you don't control, over networks you don't own, mostly has no published answers yet. You'll write some of the first ones.
FYI's
  • We work remotely across the world, with the main teams in Australia and North America. You'll need to be comfortable working across timezones.
  • Applicants must have professional-level English proficiency (written and spoken).

We are backed by Union Square Ventures and other tier-1 investors, and we are a world-class, deeply technical team of ML researchers. Pluralis is unapologetically ideological. We believe AI, and the world, end up on a better path if we succeed in implementing the protocol for intelligence.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Research Engineer - Pre-training
Research Engineer - Pre-training

OpenKrill • United States

On-site
USD 150,000 - 210,000
Remote-first culture
Visa sponsorship & relocation support
Equity opportunities
Research Scientist - Post-Training
Research Scientist - Post-Training

Pluralis-Research • San Francisco (CA)

Remote
USD 180,000 - 260,000
Equity-heavy package
Remote-first culture
Visa sponsorship
Research Scientist - Pre-training
Research Scientist - Pre-training

OpenKrill • United States

Remote
USD 180,000 - 240,000
Equity-Heavy Package
Remote-First Culture
Visa Sponsorship & Relocation support
+1
Research Engineer - Decentralized Training and Inference Verification
Research Engineer - Decentralized Training and Inference Verification

Pluralis Research • California

Remote
USD 140,000 - 210,000
Equity-heavy package
Remote-first culture
Visa sponsorship
+1
Research Scientist - Post-Training
Research Scientist - Post-Training

OpenKrill • United States

Remote
USD 180,000 - 280,000
Equity-heavy package
Remote-first culture
Visa sponsorship and relocation to US
+1
Research Scientist - Geo-Distributed Inference
Research Scientist - Geo-Distributed Inference

OpenKrill • United States

Remote
USD 180,000 - 240,000
Equity-Heavy Package
Remote-First Culture
Visa Sponsorship
+1
Research Engineer - Geo-Distributed Inference
Research Engineer - Geo-Distributed Inference

Pluralis-Research • San Francisco (CA)

Remote
USD 180,000 - 240,000
Equity-Heavy Package
Remote-First Culture
Visa Sponsorship
+1
Research Engineer - Pre-training
Research Engineer - Pre-training

Pluralis Research • California

Remote
USD 180,000 - 240,000
Equity-heavy package
Remote-first culture
Visa sponsorship
+1
Research Scientist - Geo-Distributed Inference
Research Scientist - Geo-Distributed Inference

Pluralis-Research • San Francisco (CA)

Remote
USD 180,000 - 280,000
Equity-heavy package
Remote-first culture
Visa sponsorship & relocation
+1
Research Scientist - Geo-Distributed Inference
Research Scientist - Geo-Distributed Inference

Pluralis Research • California

Remote
USD 180,000 - 240,000
Equity-heavy package
Remote-first culture
Visa sponsorship and relocation