init_ctx
|
fix _post_init_method of zero init ctx (#847)
|
2022-04-24 14:16:50 +08:00 |
shard_utils
|
[gemini] add GeminiMemoryManger (#832)
|
2022-04-24 13:08:48 +08:00 |
sharded_model
|
[gemini] polish stateful_tensor_mgr (#876)
|
2022-04-26 15:05:03 +08:00 |
sharded_optim
|
[gemini] accelerate adjust_layout() (#878)
|
2022-04-26 18:08:31 +08:00 |
sharded_param
|
[gemini] add GeminiMemoryManger (#832)
|
2022-04-24 13:08:48 +08:00 |
utils
|
[gemini] accelerate adjust_layout() (#878)
|
2022-04-26 18:08:31 +08:00 |
__init__.py
|
[gemini] add GeminiMemoryManger (#832)
|
2022-04-24 13:08:48 +08:00 |