flax
flax copied to clipboard
Debug differences in `examples/nlp_seq`.
In #1916 we saw some strange updates in the metrics when doing a no-op refactoring.
This needs more investigation, maybe also a replacement of the dataset.