Systems, devices, and methods are provided for using
model architecture search for hardware configurations. The method includes receiving, by a computer-implemented
system, information associated with a
pool of
processing units, receiving, by the computer-implemented
system, a dataset associated with a
data transformation operation, training a parameterized model associated with the
data transformation operation based on the dataset and the information associated with the
pool of
processing units, where the training includes updating at least one parameter of the parameterized model associated with configuring at least a subset of the
processing units in the
pool, and outputting one or more configurations for at least a subset of the processing units in the pool based on the training.