gpflux.layers.trackable_layer

Utility layer that tracks variables in tf.Module.

Module Contents

class TrackableLayer

Bases: gpflow.keras.tf_keras.layers.Layer

With the release of TensorFlow 2.5, our TrackableLayer workaround is no longer needed. See Prowler-io/gpflux#189. Will be removed in GPflux version 1.0.0