Olatunji Ruwase a23cda6c3b Allow modification of zero partitioned parameters (#4192) 1 年之前
..
__init__.py a23cda6c3b Allow modification of zero partitioned parameters (#4192) 1 年之前
comms_logging.py 0b507253e5 fix comm logging for inference (#4043) 1 年之前
debug.py ce535945e6 fix: change ==NONE to is (#3923) 1 年之前
exceptions.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 年之前
groups.py a855405e0b DeepSpeed Ulysses release (#4198) 1 年之前
init_on_device.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 年之前
logging.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 年之前
mixed_precision_linkage.py a23cda6c3b Allow modification of zero partitioned parameters (#4192) 1 年之前
numa.py 5dadf68771 support HBM in utils/numa.py (#3918) 1 年之前
nvtx.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 年之前
tensor_fragment.py a23cda6c3b Allow modification of zero partitioned parameters (#4192) 1 年之前
timer.py b354c28b76 polishing timers and log_dist (#3996) 1 年之前
types.py 0a61d5d664 Hybrid Engine Refactor and Llama Inference Support (#3425) 1 年之前
zero_to_fp32.py 241ae39a29 zero_to_fp32 script adds support for tag argument (#4089) 1 年之前