Model

A deterministic model from inputs to outputs. You can choose among several built-in models, or bring in your own model by overwriting Module. The model forward pass is orchestrated by a model manager.

Contents:

• Built-in models
• Flax Linen Module
• Model manager
• Model Utils