flax

[flax:examples:nlp_seq] Create a main.py file to run tests with config files to match other examples. #jax-fixit

#5126

Merged