File size: 268 Bytes
5fa1a76 |
1 2 3 4 |
Main concepts The library is built around three types of classes for each model: Model classes can be PyTorch models (torch.nn.Module), Keras models (tf.keras.Model) or JAX/Flax models (flax.linen.Module) that work with the pretrained weights provided in the library. |