microsoft/DeepSpeed

Pull Requests Commits

Install torchvision too

loadams committed 330 days ago

7a6b7220

loadams committed 330 days ago

b551b582

loadams committed 330 days ago

d2e52e32

Ensure we get torch+cu vs torch+cpu

loadams committed 331 days ago

289871fd

Turn off NCCL debug

loadams committed 331 days ago

2c8488b7

Pin torch in nv-ds-chat workflow

loadams committed 331 days ago

6d57a4a2

generalize deepspeed linear and implement it for non cuda systems (#6932)

oelayan7 committed 331 days ago

Verified 593de925

Update torch versions to support 2.6 (#6977)

loadams committed 332 days ago

Verified 8ad48725

Add CUDA 12.8 support and comment on CUDA 12.7 (#6975)

loadams committed 333 days ago

Verified 46c6c9e6

Update build_win.bat script to exclue GDS op as it lacks Windows support. (#6971)

loadams committed 336 days ago

Verified 1640f6df

Precisely track nvme optimizer offload (#6963)

tjruwase committed 337 days ago

Verified 470dd6dc

Update version.txt after 0.16.3 release (#6965)

loadams committed 339 days ago

Verified de4596be

Using explicit GPU upcast for ZeRO-Offload (#6962)

xylian86 committed 339 days ago

Verified c17dc33c

Update `torch.norm` to `torch.linalg.norm` and `torch.linalg.vector_norm` (#6931)

loadams committed 339 days ago

Verified 8d1bc0a0

Add the missing view operations from sequence parallel(async). (#6750)

inkcherry committed 339 days ago

Verified bc76b04e

Remove Duplicate Declaration of pandas in `Dockerfile` (#6959)

Zerohertz committed 343 days ago

Verified 7f3d669b

Update import for torchvision.transformers (#6958)

loadams committed 343 days ago

Verified f97f0885

Add extra_repr to Linear classes for debugging purpose (#6954)

Xia-Weiwen committed 344 days ago

Verified 018ece5a

`warn` to `warning` (#6952)

qgallouedec committed 345 days ago

Verified 05eaf3d1

[inf] Add config var to enable keeping module on host (#6846)

oelayan7 committed 345 days ago

Verified fae714d6

Pin nv-a6000 workflow (#6938)

loadams committed 347 days ago

Verified 66d3d3e9

Remove op compilation flags due to perf issue (#6944)

NirSonnenschein committed 347 days ago

Verified 396f8db7

Support pure meta model lm_head tp (#6812)

Yejing-Lai committed 350 days ago

Verified fa8db5cf

Add information on security expectations with this software (#6941)

loadams committed 351 days ago

Verified 1d15ef0a

Add position_ids arg to OPTEmbedding forward function (#6939)

lekurile committed 351 days ago

Verified 0fc3daad

Add deepseek autotp (#6937)

Yejing-Lai committed 351 days ago

Verified 45fce45c

Fix windows blog examples (#6934)

loadams committed 352 days ago

Verified 53fb5795

Fix building on Windows with presence of Triton (#6749)

woct0rdho committed 352 days ago

Verified b62c84d8

Update python version classifiers (#6933)

loadams committed 352 days ago

Verified 6628127a

Use `torch.log1p` (#6930)

kit1980 committed 353 days ago

Verified c41b0c28

Older