feat: use model definition

fix(inference): update flax version (temporary fix)

Merge pull request #93 from borisdayma/inference

feat: update samples

feat: update samples

feat: rename variables

feat: reorganize samples

fix: duplicate samples

fix: OOM

fix: OOM with checkpoints

fix: comment

feat: cleanup training script

feat: more samples

feat: add customization parameters

feat: reorganize samples

feat: add more samples

feat: update wandb inference

Merge pull request #91 from borisdayma/feat-inf

feat: more samples

feat: cleanup

feat: add sample

fix: typo

fix: pmap clip32

feat: cleanup

feat: allow latest version only

refactor: loop over runs

feat: create a table

feat: add scoring

feat: remove cache before creating artifacts

feat: add functions

feat: log to backend

fix: state.step type

Merge branch 'main' of https://github.com/borisdayma/dalle-mini

feat: get rid of global_step + log more metrics

fix(seq2seq): memory issue

feat: update scripts

feat: use optax for gradient accumulation

feat: no need for default values

feat: limit artifacts size

feat: log epoch + check params

feat: update defaults

fix(seq2seq): normalize text

fix(seq2seq): use streaming arg

fix: remove breakpoint

feat: handle streaming

Merge branch 'main' of https://github.com/borisdayma/dalle-mini into fix-opt_state

feat(inference_notebook): dalle-mini is installable

Merge pull request #74 from borisdayma/feat-setup

Replace notebooks with the correct versions.

Notebooks that demonstrate streaming encoding