Experiment aimed at cleaner style for writing complex models.
Experiment aimed at cleaner style for writing complex models.

Ideas:
  - Highlight the top-level composition of layers making up the model; aim for
    visual clarity of the code.
  - Factor out the many parameters by means of intermediate variables and
    internal functions.
  - Use Branch(...) rather than Dup(...), Parallel(...)
  - Use Select(...) for arg rearrangements, in place of Parallel(...),
    Drop(...), Swap(...), ...
  - Have private functions return layer lists without wrapping them in
    Serial(...). This reduces unnecessary nesting of Serial in Serial.
PiperOrigin-RevId: 284275946

google/trax,google/trax