microsoft / microsoft/aurora

Make the number of ensemble members internal to the Aurora model.

Open
#192 2 comments 0 reactions 0 assignees View on GitHub

@Slowika is already working on this.

Since Aug 13, 2026.

  • #199 by @Slowika — open
Dominant language
Python
Stars
1k
Forks
174
PR merge metrics
No merged PRs in 30d

Description

Aurora 1.5 Ensemble has the option of being based on an ensemble with N members. This is currently achieved by running the model N times in the context in which the ensemble approach is required.

There should be an option to make the ensemble members internal to Aurora, so that getting all of the ensemble results is done using a single call to e.g. forward, as opposed to running N of them in a loop.

This would expose more opportunities for making the full use of GPUs. Currently, each ensemble member is a completely separate model with separate inputs and outputs. This means that a high-performance GPU may be under-utilised compared to representing all ensemble members using the same tensor.

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by tracing the Aurora 1.5 Ensemble model and its forward entry point to understand how ensemble members are currently invoked. Confirm how inputs and outputs are represented across the existing repeated calls, then define the internal ensemble interface. Done means one forward call returns all ensemble results while enabling the members to use the GPU more fully.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Active
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.