Как избежать больших объектов в графе задач

Я запускаю симуляции с использованием dask.distributed. Моя модель определена в функции с задержкой, и я складываю несколько реализаций. Упрощенная версия того, что я делаю, приведена в этом фрагменте кода:

import numpy as np
import xarray as xr
import dask.array as da
import dask
from dask.distributed import Client
from itertools import repeat 

@dask.delayed
def run_model(n_time,a,b):
    result = np.array([a*np.random.randn(n_time)+b])
    return result

client = Client()

# Parameters
n_sims = 10000
n_time = 100
a_vals = np.random.randn(n_sims)
b_vals = np.random.randn(n_sims)
output_file = 'out.nc'

# Run simulations
out = da.stack([da.from_delayed(run_model(n_time,a,b),(1,n_time,),np.float64) for a,b in zip(a_vals, b_vals)])

# Store output in a dataframe
ds = xr.Dataset({'var1': (['realization', 'time'], out[:,0,:])},
             coords={'realization': np.arange(n_sims),
                     'time': np.arange(n_time)*.1})

# Save to a netcdf file -> at this point, computations will be carried out
ds.to_netcdf(output_file)

Если я хочу запустить много симуляций, я получаю следующее предупреждение:

/home/user/miniconda3/lib/python3.6/site-packages/distributed/worker.py:840: UserWarning: Large object of size 2.73 MB detected in task graph: 
  ("('getitem-32103d4a23823ad4f97dcb3faed7cf07', 0,  ... cd39>]), False)
Consider scattering large objects ahead of time
with client.scatter to reduce scheduler burden and keep data on workers

    future = client.submit(func, big_data)    # bad

    big_future = client.scatter(big_data)     # good
    future = client.submit(func, big_future)  # good
  % (format_bytes(len(b)), s))

Насколько я понимаю (из этого и этого вопроса), метод, предложенный предупреждением, помогает получать большие данные в функцию. Однако мои входы - это все скалярные значения, поэтому они не должны занимать почти 3 МБ памяти. Даже если функция run_model() не принимает никакого аргумента вообще (поэтому никакие параметры не передаются), я получаю такое же предупреждение.

Я также просмотрел график задачи, чтобы узнать, есть ли какой-то шаг, требующий загрузки большого количества данных. Для трех реализаций это выглядит так:

Поэтому мне кажется, что каждая реализация обрабатывается отдельно, которая должна содержать объем данных, чтобы справиться с минимумом.

Я хотел бы понять, что на самом деле является шагом, который создает большой объект и что мне нужно сделать, чтобы разбить его на более мелкие части.