.. |
__init__.py
|
541e423ae6
Enable tensor fragments for zero 2 & 3 (#2727)
|
1 年之前 |
comms_logging.py
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |
debug.py
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |
exceptions.py
|
c0af6d90f7
Refactor MoE and Groups API to simplify model creation and mangement (#1798)
|
2 年之前 |
groups.py
|
5fe9d61065
Tensor parallelism for Mixture of Experts (#2074)
|
2 年之前 |
init_on_device.py
|
aa88137b8d
Add Inference support for running the BigScience-BLOOM Architecture (#2083)
|
2 年之前 |
logging.py
|
50a49e42fb
[logger] implement warning_once (#3021)
|
1 年之前 |
mixed_precision_linkage.py
|
541e423ae6
Enable tensor fragments for zero 2 & 3 (#2727)
|
1 年之前 |
nvtx.py
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |
tensor_fragment.py
|
541e423ae6
Enable tensor fragments for zero 2 & 3 (#2727)
|
1 年之前 |
timer.py
|
98cc35b6a8
Abstract accelerator (step 3) (#2677)
|
1 年之前 |
types.py
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 年之前 |
zero_to_fp32.py
|
541e423ae6
Enable tensor fragments for zero 2 & 3 (#2727)
|
1 年之前 |