Making large AI models cheaper, faster and more accessible
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
LuGY 839847b7d7
[zero]support zero2 with gradient accumulation (#4511)
1 year ago
..
components_to_test
kit
test_amp
test_analyzer
test_auto_parallel
test_autochunk
test_booster
test_checkpoint_io
test_cluster
test_comm
test_config
test_context
test_data
test_data_pipeline_tensor_parallel
test_device
test_engine
test_fx
test_kernels
test_layers
test_lazy
test_moe
test_optimizer
test_pipeline
test_shardformer
test_tensor
test_trainer
test_utils
test_zero [zero]support zero2 with gradient accumulation (#4511) 1 year ago
__init__.py