Code

Flax sharded#17760

Code

Merged

ArthurZucker merged 18 commits intohuggingface:mainhuggingface/transformers:mainfrom ArthurZucker:flax-shardedArthurZucker/transformers:flax-shardedCopy head branch name to clipboardJun 22, 2022

+305-22

Commits

Commits on Jun 17, 2022

• initial commit

  ArthurZucker

  committed
  cc773a2

• style

  ArthurZucker

  committed
  62df3a1

• update

  ArthurZucker

  committed
  ea656fc

• fix flax test

  ArthurZucker

  committed
  81e7ab8

• functions in hub

  ArthurZucker

  committed
  e62d65b

Commits on Jun 18, 2022

• Update

  ArthurZucker

  committed
  db435a1

• fix bug

  ArthurZucker

  committed
  2fba0be

Commits on Jun 20, 2022

• Update and fix test

  ArthurZucker

  committed
  d631191

• update and fix test, now running

  ArthurZucker

  committed
  a664889

Commits on Jun 21, 2022

• Apply suggestions from code review

  Show description for b96949e
  

  3 peopleauthored
  b96949e

• update based on review

  ArthurZucker

  committed
  24670f0

• Merge branch 'flax-sharded' of https://github.com/ArthurZucker/transformers into flax-sharded

  ArthurZucker

  committed
  cd89fa5

• style

  ArthurZucker

  committed
  5f0b8c3

• clean

  ArthurZucker

  committed
  111688f

• remove conversion to np array

  ArthurZucker

  committed
  83ec388

• Merge branch 'main' of https://github.com/huggingface/transformers into flax-sharded

  ArthurZucker

  committed
  1d48275

• Merge branch 'main' of https://github.com/huggingface/transformers into flax-sharded

  ArthurZucker

  committed
  128b666

• Merge branch 'main' into flax-sharded

  ArthurZucker

  authored
  3261abf