.. |
activation_checkpointing
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |
checkpoint_engine
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |
comm
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |
compression
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |
data_pipeline
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |
fp16
|
80d8fcbdb3
Improve overflow handling (#2944)
|
1 年之前 |
pipe
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |
swap_tensor
|
98cc35b6a8
Abstract accelerator (step 3) (#2677)
|
1 年之前 |
zero
|
80d8fcbdb3
Improve overflow handling (#2944)
|
1 年之前 |
__init__.py
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |
bf16_optimizer.py
|
541e423ae6
Enable tensor fragments for zero 2 & 3 (#2727)
|
1 年之前 |
config.py
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |
config_utils.py
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |
constants.py
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |
dataloader.py
|
7c99def0f0
Data efficiency library update (#2866)
|
1 年之前 |
eigenvalue.py
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |
engine.py
|
f1d2a15b50
better eval sampler (#2907)
|
1 年之前 |
lr_schedules.py
|
316c4a43e0
Add flake8 to pre-commit checks (#2051)
|
2 年之前 |
progressive_layer_drop.py
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |
quantize.py
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |
sparse_tensor.py
|
fcb3ca5e66
Proposal of how we might use sparse tensors for gradients (#1484)
|
3 年之前 |
state_dict_factory.py
|
ccb8eb81fb
Add checkpoint sharding unit tests (#2561)
|
1 年之前 |
utils.py
|
d3de737550
Remove deprecated `torch._six` imports (#2863)
|
1 年之前 |
weight_quantizer.py
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |