Repository navigation

Commits

History for

onmain

Commits on Mar 14, 2022
- feat: support pod (#139)
  borisdayma
  authored
  803ccbf
Commits on Feb 23, 2022
- fix: no gradient checkpointing for new model
  borisdayma
  committed
  2e02683
- feat: no gradient checkpointing for params init
  borisdayma
  committed
  b798ed3
- feat: update configs
  borisdayma
  committed
  79557f9
- fix(train): consider schedule offset
  borisdayma
  committed
  bc4734f
- feat(train): local jax cache
  borisdayma
  committed
  9f5e879
- feat: add bucket reference to artifact
  borisdayma
  committed
  d368fb6
Commits on Feb 22, 2022
- style: lint
  borisdayma
  committed
  d5d442a
- feat: handle gradient checkpointing
  borisdayma
  committed
  5173ec7
- feat: load from bucket
  borisdayma
  committed
  1c4e839
- feat(train): save to bucket
  borisdayma
  committed
  50498e6
- feat: reduce artifact space + offset step
  borisdayma
  committed
  34cf91c
Commits on Feb 21, 2022
- feat(demo): update reference
  borisdayma
  committed
  e558000
- feat: restore weights on CPU
  borisdayma
  committed
  5f954fc
- feat(train): simplify tokenizer loading
  borisdayma
  committed
  4cb21dd
Commits on Feb 10, 2022
- feat: cleanup notebook
  borisdayma
  committed
  5a390e8
- feat(demo): uncomment pip install
  borisdayma
  committed
  094e178
- feat: improve inference demo
  borisdayma
  committed
  35fe578
Commits on Feb 7, 2022
- feat(train): use compilation cache
  borisdayma
  committed
  da9367c
Commits on Feb 3, 2022
- fix: style
  borisdayma
  committed
  d483294
Commits on Feb 2, 2022
- Merge branch 'main' of https://github.com/borisdayma/dalle-mini into main
  borisdayma
  committed
  0a691de
- feat: log num_parameters early
  borisdayma
  committed
  7cfe576
- fix: distributed shampoo class
  borisdayma
  committed
  696422e
- feat: update distributed_shampoo
  borisdayma
  committed
  5996680
Commits on Feb 1, 2022
- feat(train) - handle multiple nodes (#130)
  borisdayma
  authored
  0952927
- feat: handle model parallel
  borisdayma
  committed
  1bb3269
Commits on Jan 31, 2022
- feat(train): more custom x-axis
  borisdayma
  committed
  5f28cd2
Commits on Jan 29, 2022
- fix(train): opt_state_shape for distributed_shampoo
  borisdayma
  committed
  225b6ff
Commits on Jan 27, 2022
- feat(train): split artifact into model/state
  borisdayma
  committed
  fa5b058
Commits on Jan 26, 2022
- feat(train): another 25% faster
  borisdayma
  committed
  14abe8c
- feat(train): overhead from 70% to 1% 🥳
  borisdayma
  committed
  2b7f5f1
- feat(pjit): follow t5x style
  borisdayma
  committed
  7b5868f
Commits on Jan 25, 2022
- fix(train): grads spec
  borisdayma
  committed
  00710bc
- feat(train): improve pjit speed
  borisdayma
  committed
  f254058
- fix(train): consider correct batch size
  borisdayma
  committed
  b7c7458

Breadcrumbs

User selector

Datepicker

Commit history

Commits on Mar 14, 2022

Commits on Feb 23, 2022

Commits on Feb 22, 2022

Commits on Feb 21, 2022

Commits on Feb 10, 2022

Commits on Feb 7, 2022

Commits on Feb 3, 2022

Commits on Feb 2, 2022

Commits on Feb 1, 2022

Commits on Jan 31, 2022

Commits on Jan 29, 2022

Commits on Jan 27, 2022

Commits on Jan 26, 2022

Commits on Jan 25, 2022