Jiarui Fang f7e276fa71 [Gemini] add GeminiAdamOptimizer (#1960)		2022-11-16 14:44:28 +08:00
..
amp	[NFC] polish colossalai/amp/naive_amp/__init__.py code style (#1905)	2022-11-11 17:49:18 +08:00
auto_parallel	[autoparallel] remove redundancy comm node (#1893)	2022-11-15 10:53:41 +08:00
builder	…
cli	…
communication	…
context	updated tp layers	2022-11-02 12:19:38 +08:00
device	…
engine	…
fx	[SC] add GPT example for auto checkpoint (#1889)	2022-11-11 23:17:25 +08:00
gemini	MemStatsCollectorStatic (#1765)	2022-11-07 16:49:03 +08:00
kernel	updated flash attention api	2022-11-15 15:25:39 +08:00
logging	fixed logger	2022-11-15 16:00:07 +08:00
nn	[Gemini] add GeminiAdamOptimizer (#1960)	2022-11-16 14:44:28 +08:00
pipeline	…
registry	…
tensor	[autoparallel] remove redundancy comm node (#1893)	2022-11-15 10:53:41 +08:00
testing	…
trainer	…
utils	[ColoTensor] reconfig ColoInitContext, decouple default_pg and default_dist_spec. (#1953)	2022-11-15 16:24:16 +08:00
zero	[Gemini] add GeminiAdamOptimizer (#1960)	2022-11-16 14:44:28 +08:00
__init__.py	version to 0.1.11rc2 (#1832)	2022-11-08 17:25:15 +08:00
constants.py	updated tp layers	2022-11-02 12:19:38 +08:00
core.py	…
global_variables.py	updated tp layers	2022-11-02 12:19:38 +08:00
initialize.py	…