PR #1920 Improves seq2seq example

Factors out models and input pipeline for seq2seq example

marcvanzee committed 4 years ago

Incorporate changes from code review

marcvanzee committed 4 years ago

Fix types

marcvanzee committed 4 years ago

Update docstrings in seq2seq examples

marcvanzee committed 4 years ago

Merge branch 'main' of https://www.github.com/google/flax into improve-seq2seq

marcvanzee committed 4 years ago

flax Improves seq2seq example #1920 Merged