Mikhail Druzhinin 4bf4ab7ac5 Fix partial recovery of sparse_tensor_module_names and dynamically check if gradient data is sparse (#1562) 2 years ago
..
autotuning bda3d0e6b9 Add autotuning news post (#1565) 2 years ago
elasticity 85ce85dd5f Remove redundant pass (#1509) 3 years ago
inference 9ce00a2171 Tensor-Parallelism general support (#1512) 2 years ago
launcher 9caa74e577 Autotuning (#1554) 2 years ago
module_inject 9ce00a2171 Tensor-Parallelism general support (#1512) 2 years ago
moe 2887349cd4 Adding Tutel to MoE layer (#1528) 3 years ago
ops 3ed77304df Fix sparse attention for small block-sizes (#1545) 2 years ago
pipe 65c2f974d8 Pipeline parallel training engine. (#392) 4 years ago
profiling 2c62d657a4 typo in profiler.py (#1527) 3 years ago
runtime 4bf4ab7ac5 Fix partial recovery of sparse_tensor_module_names and dynamically check if gradient data is sparse (#1562) 2 years ago
utils 9caa74e577 Autotuning (#1554) 2 years ago
__init__.py 9ce00a2171 Tensor-Parallelism general support (#1512) 2 years ago
constants.py ed3de0c21b Quantization + inference release (#1091) 3 years ago
env_report.py ca9ab1201f ds_report bug fix on cpu and guard torch import in setup.py (#524) 4 years ago
git_version_info.py 6380ee3511 Fixes for RTD build errors (#606) 3 years ago