5fa1a76
1
In addition, all models are also either a torch.nn.Module, tf.keras.Model or flax.linen.Module subclass.
