1. 07 Jul, 2021 3 commits
    • Kevin Canwen Xu's avatar
    • Daniel Stancl's avatar
      [Flax] Add FlaxMBart (#12236) · 61400e1e
      Daniel Stancl authored
      
      
      * Copy BART to MBart and rename some stuff
      
      * Add copy statements pointing to FlaxBart
      
      * Update/add some common files
      
      * Update shift_tokens_rigth + fix imports
      
      * Fix shift_tokens_right method according to MBart implementation
      
      * Update shift_tokens_right in tests accordingly
      
      * Fix the import issue and update docs file
      * make style quality
      
      * Do some minor changes according to patil-suraj suggestions
      
      * Change the order of normalization layer and attention
      
      * Add some copu statementes
      
      * Update generate method and add integration test for mBart
      
      * Make a few updates after a review
      
      Besides, add `lang_code_to_id` to MBartTokenizeFast
      
      * fix-copies; make style quality
      
      * Apply suggestions from code review
      
      * Apply suggestions from code review
      
      * Apply suggestions from code review
      
      * fix output type, style
      
      * add copied from
      
      * resolve conflicts
      Co-authored-by: default avatarSuraj Patil <surajp815@gmail.com>
      61400e1e
    • Suraj Patil's avatar
      [examples/flax] add adafactor optimizer (#12544) · 2d42915a
      Suraj Patil authored
      
      
      * add adafactor
      
      * Update examples/flax/language-modeling/run_mlm_flax.py
      Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
      2d42915a
  2. 06 Jul, 2021 9 commits
  3. 05 Jul, 2021 12 commits
  4. 04 Jul, 2021 1 commit
  5. 02 Jul, 2021 7 commits
  6. 01 Jul, 2021 8 commits