ML Researcher

Base Compute; Melbourne; On-site; AUD 150,000 - 210,000; Full time

Benefits offered by this job

Equity

Hybrid work (occasional in-office)

Direct influence on technical方向

Job summary

Base Compute, an on-device AI lab based in Melbourne, seeks an ML Researcher to push the frontier of on-device AI. You will identify problems, design experiments and derive insights that translate into real-world performance improvements.

You’ll own parts of the research agenda and influence the company’s technical bets. You’ll work on inference research, model routing between on-device and frontier models, and autoresearch pipelines to accelerate the R&D loop while building rigorous evaluations

Qualifications

• PhD in ML or equivalent industry research experience.
• Deep understanding of LLM architectures and AI inference.
• Expertise in speculative decoding, quantization theory, distillation, reinforcement learning.
• Proven track record: papers, open-source projects or blogs.
• Strong communication and reproducible documentation.

Responsibilities

• Identify and validate on-device efficiency approaches via experiments.
• Build and evaluate model routing between on-device and frontier API models.
• Design autonomous pipelines to explore and evaluate research ideas.
• Develop rigorous real-world performance evaluations and benchmarks.

Skills

PhD in ML

LLM architectures

Inference optimization

Research dissemination

Education

PhD in ML or equivalent

Tools

CUDA

ROCm

Metal

Triton

Job description

About Us

Base Compute is an AI inference lab. Our mission is to bring AGI on-device. We believe in a world where everyone has access to intelligence: fast, private and always available on your device.

We’re building the infrastructure for the next generation of on-device AI, from silicon-level optimizations to distributed inference systems.

We’re working on hard problems at the intersection of inference efficiency, model intelligence and autonomous research.

The Role

We’re looking for an ML Researcher to work at the frontier of on-device AI. This role is for someone who identifies problems and potentials, designs and executes experiments and derives insights that translate into real-world performance.

You’ll have significant ownership over our research agenda and direct influence on the technical bets the company makes.

What You’ll Work On

• Inference research: Identifying and validating new approaches to on-device efficiency, including speculative decoding variants, novel quantization schemes and entirely new techniques yet to be discovered


• Model routing research: Building the intelligence that decides how requests are served between on-device vs. frontier API models


• Autoresearch pipelines: Designing systems that can autonomously explore, hypothesize and evaluate research ideas that accelerate our R&D loop


• Evaluations and benchmarks: Developing rigorous evals that measure performance in the real world, outside of clean academic settings

What We’re Looking For

• PhD in ML or equivalent industry research experience


• Deep understanding of LLM architectures and the principles of AI inference


• Expertise in a relevant topic, such as speculative decoding, quantization theory, model distillation, reinforcement learning


• A track record of producing results that people build on: research papers, open-source projects or blog posts that prove out novel ideas


• Good communication: the ability to explain complex ideas simply, give honest feedback and document findings in a reproducible way


• Nice-to-haves:


• Familiarity with GPU and accelerator architectures and kernel optimization (CUDA, ROCm, Metal, Triton, etc.)


• Experience deploying models under on-device constraints (memory bandwidth, latency budgets, and thermal and power ceilings)

What We Offer

• Equity and strong base salary


• Direct influence on technical direction: your ideas will shape the roadmap


• Work on genuinely hard problems that haven't been solved yet


• Small team, fast iteration, low bureaucracy

Location

• The team is based in Melbourne and Berlin and works in-person from the office most days. We require strong written and spoken English, since the team collaborates across time zones.