.. |
__init__.py
|
65ab64481f
Add API for updating ZeRO gradients (#6590)
|
1 周之前 |
bwc.py
|
08e0733e4a
Support MoE for pipeline models (#5338)
|
6 月之前 |
comms_logging.py
|
0b507253e5
fix comm logging for inference (#4043)
|
1 年之前 |
config.py
|
995ba11928
Add throughput timer configuration (#5363)
|
5 月之前 |
debug.py
|
40342055ce
Remove hooks on gradient accumulation on engine/optimizer destroy (#4858)
|
9 月之前 |
exceptions.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
groups.py
|
8b191d7ccf
Long sequence parallelism (Ulysses) integration with HuggingFace (#5774)
|
2 月之前 |
init_on_device.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
logging.py
|
ce468c3756
add option to disable logger while compiling to avoid graph breaks (#6496)
|
6 天之前 |
mixed_precision_linkage.py
|
65ab64481f
Add API for updating ZeRO gradients (#6590)
|
1 周之前 |
numa.py
|
662a421b05
Safe usage of popen (#6490)
|
1 月之前 |
nvtx.py
|
b421e8c8f3
Disable nvtx decorator to avoid graph break (#5697)
|
3 月之前 |
tensor_fragment.py
|
65ab64481f
Add API for updating ZeRO gradients (#6590)
|
1 周之前 |
timer.py
|
f74ea69abf
Improve DS logging control (#6602)
|
1 周之前 |
torch.py
|
6dcced1d5c
Cleanup required_torch_version code and references. (#5370)
|
6 月之前 |
types.py
|
0a61d5d664
Hybrid Engine Refactor and Llama Inference Support (#3425)
|
1 年之前 |
z3_leaf_module.py
|
19e0dc39ba
Delay reduce-scatter for ZeRO3 leaf modules (#5008)
|
8 月之前 |
zero_to_fp32.py
|
bf60fc0ca6
Support safetensors export (#6579)
|
1 周之前 |