**Improvements/fixes**
TF datasets are now passed to DeepEnsemble.model.fit. This change enables support for training over a specified number of steps (rather than a set number of epochs) for small datasets, which may require data repetition to achieve the desired step count. https://github.com/secondmind-labs/trieste/pull/907
An optional predict_fn parameter has been added to IndependentReparametrizationSampler.__init__. This parameter allows generating samples where the mean and variance come from sources other than model.predict. This feature is particularly useful for drawing samples from models that separate epistemic and aleatoric uncertainty, providing greater flexibility in controlling the sources of uncertainty in the generated samples. https://github.com/secondmind-labs/trieste/pull/903