Jiarui Fang 59bf2dc590 [zero] initialize a stateful tensor manager (#614)

2022-04-06 16:18:49 +08:00

..

__init__.py

[zero] polish sharded param name (#484)

2022-03-22 14:36:16 +08:00

sharded_param.py

[zero] initialize a stateful tensor manager (#614)

2022-04-06 16:18:49 +08:00

sharded_tensor.py

[zero] label state for param fp16 and grad (#551)

2022-03-30 15:57:46 +08:00

tensorful_state.py

[zero] label state for param fp16 and grad (#551)

2022-03-30 15:57:46 +08:00