Pluralis Research

Machine Learning Engineer - Intern

Pluralis Research · Melbourne, Victoria, Australia

Technology, Information and Internet · 2-10 employees

14 h ago
Remote Junior (0-2 yrs) Full-time Australia
Log in to apply, save this posting, or score it against your profile with AI.

About the role

Design and implement components for the Agora decentralized training system while working with large-scale training infrastructure. You will contribute to the production codebase daily through code reviews, standups, and collaborative pairing.

What they look for

Machine Learning Python PyTorch Distributed Systems Multiprocessing Async I/O Threading Cloud Infrastructure AWS GCP Protocol Learning Large-scale Training Parallel Systems Networking

Requirements

Candidates must be currently enrolled in or have recently completed a Masters or PhD in machine learning, computer science, or a related field. Strong proficiency in Python, PyTorch, and experience with distributed systems and cloud platforms are required.

Full description

Pluralis Research is pioneering Protocol Learning - a fully decentralised way to train and deploy AI models that opens this layer to individuals rather than well resourced corporates. By pooling compute from many participants, incentivising their efforts, and preventing any single party from controlling a model's full weights, we're creating a genuinely open, collaborative path to frontier-scale AI.

As a Machine Learning Engineer Intern, you'll work on Agora - our production decentralised training system - alongside the engineers running real multi-node training at scale. This is a fixed-term internship (3 months, with option to extend). Intern projects are well-scoped pieces of our live roadmap, not side experiments: you'll ship real work in week one and own a genuine open problem by the end of your internship.

Key Responsibilities

  • Design, build and ship a well-scoped project on the Agora roadmap, with a final presentation to both the engineering and research teams.
  • Build and improve concurrent and parallel systems components (multiprocessing, async I/O and threading) within a production distributed training stack.
  • Work hands-on with large-scale training infrastructure across cloud providers (AWS/GCP).
  • Contribute to the production codebase daily: code review, standups and pairing with a buddy on the Agora team.

What We're Looking For

  • Current enrolment in (or recent completion of) a Masters or PhD in machine learning, computer science or a related field. Engineering-inclined candidates from either level are welcome.
  • A genuine ML background. You can keep pace with a research-driven team, not just a strong generalist engineering profile.
  • Strong Python and PyTorch.
  • Experience building concurrent or parallel systems (multiprocessing, async I/O, threading).
  • Hands-on exposure to distributed machine learning, via internship, coursework or serious projects.
  • Experience working with AWS, GCP or other hyperscalers.
  • Australia-based.

Nice to Have

  • Top-tier ML publications (welcome, but not required for systems-focussed candidates).
  • Open-source contributions to ML frameworks, distributed systems or networking libraries.

FYI's

  • This internship is for Australia-based candidates only.
  • Applicants must have professional-level English proficiency (written and spoken).
  • Pluralis is a remote team across Australia and the US. You'll need to be comfortable working across timezones and collaborating with a diverse, distributed group.
  • Recruiters: we aren't looking for agency support at this time. We'll reach out if we need help.

Backed by Union Square Ventures and other tier-1 investors, we're a world-class, deeply technical team of ML researchers. Pluralis is unapologetically ideological. We view the world as a better place if we are able to implement what we are attempting, and Protocol Learning as the only plausible approach to preventing a handful of massive corporations monopolising model development, access and release, and achieving massive economic capture. If this resonates, please apply.