[Flax] Add flax marian (#12595)
* fix_torch_device_generate_test
* remove @
* add marian
* finish make style
* add model
* add docs
* add test
* add integration tests
* up
* solve bug
* correct tests
* correct some tests
* Apply suggestions from code review
Co-authored-by: Suraj Patil <surajp815@gmail.com>
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
* correct adapt marian
* finish
Co-authored-by: Patrick von Platen <patrick@huggingface.co>
Co-authored-by: Suraj Patil <surajp815@gmail.com>
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
