microsoft/DeepSpeed

Pull Requests Commits

add bwd support for list/dict types returned in fwd

jeffra committed 3 years ago

8aeab68d

port over engine changes

jeffra committed 3 years ago

ae988177

turn off prints

jeffra committed 3 years ago

e33f5e98

progress on latest master

jeffra committed 3 years ago

d05ca4fe

[ZeRO] Default disable elastic ckpt in stage 1+2 and reduce CPU memory overhead during ckpt load (#1525)

jeffra committed 3 years ago

Verified 3293cf72

force clear stashed tensors (#1698)

jeffra committed 3 years ago

Verified e4cf40d6

support module and no python args for launcher (#1690)

liamcli committed 3 years ago

Verified fead387f

add -lcurand to fix torch-nightly issue w. JIT (#1688)

jeffra committed 3 years ago

Verified a85dce07

[docs] switch to transparent dark logo

jeffra committed 3 years ago

3a4cb042

fix the half-precision version of rotary_pos_emb kernel (#1683)

RezaYazdaniAminabadi committed 3 years ago

Verified 762e697a

GPT-J inference support (#1670)

RezaYazdaniAminabadi committed 3 years ago

Verified 289c3f9b

[docs] add gh-dark-mode logo

jeffra committed 3 years ago

7e857aab

[docs] add light-mode logo

jeffra committed 3 years ago

Verified 9c5cf3a5

add segfault guard for cpu-adam/adagrad (#1681)

jeffra committed 3 years ago

Verified 2422ec48

Copy grads to cpu in z1-offload (#1679)

tjruwase committed 3 years ago

Verified cef116f8

[docs] add logo (#1676)

jeffra committed 3 years ago

Verified c2735996

update results about public Pile dataset (#1675)

conglongli committed 3 years ago

Verified aca64799

Fix largest param numel calculation (#1623)

tjruwase committed 3 years ago

Verified 4354c3cc

support CPU Adam and Adagrad on Windows with SDK 10.0.22000 (#1634)

lostmsu committed 3 years ago

Verified 74493b2b

jeffra committed 3 years ago

b6f0ac97

Various small documentation text improvements (#1665)

manuelciosici committed 3 years ago

Verified d0ab7224

Convert the fp16_params to group of parameters (#1651)

RezaYazdaniAminabadi committed 3 years ago

Verified 559c4ce1

Replace calls to print() with calls to logger (#1664)

manuelciosici committed 3 years ago

Verified 40ce131c

[save_fp16_model] return status (#1663)

stas00 committed 3 years ago

Verified 317400ea

follow-up to #1652, resolved a100-80gb issue (#1655)

jeffra committed 3 years ago

Verified d93d924a

add backup cpu-arch detection if py-cpuinfo fails (#1652)

jeffra committed 3 years ago

Verified cbd68dc4

New feature contribution guideline (#1646)

conglongli committed 3 years ago

Verified 752319c7

Add torchvision to requirements-dev.txt (#1642)

aphedges committed 3 years ago

Verified 8bbf081a

Updating autotuner readme file to add hyperparameter adjustment suggestions (#1641)

minjiaz committed 3 years ago

Verified f2c433f0

Fix cpu-adam AVX performance (#1637)

RezaYazdaniAminabadi committed 3 years ago

Verified 259936a7

Older