ColossalAI/colossalai
Frank Lee edb67cb378
[autoparallel] refactored the data structure for sharding strategy (#1610)
2022-09-20 11:20:54 +08:00
..
amp
auto_parallel [autoparallel] refactored the data structure for sharding strategy (#1610) 2022-09-20 11:20:54 +08:00
builder [NFC] polish colossalai/builder/__init__.py code style (#1560) 2022-09-08 22:11:04 +08:00
cli
communication
context [autoparallel] refactored shape consistency to remove redundancy (#1591) 2022-09-13 18:30:18 +08:00
device [tensor]add 1D device mesh (#1492) 2022-08-25 16:48:12 +08:00
engine [engin/schedule] use p2p_v2 to recontruct pipeline_schedule (#1408) 2022-08-12 11:33:26 +08:00
fx [fx] Add pofo solver (#1608) 2022-09-20 11:20:48 +08:00
gemini [NFC] polish colossalai/gemini/update/chunkv2.py code style (#1565) 2022-09-08 22:11:04 +08:00
kernel
logging
nn [embedding] updates some default parameters 2022-09-15 15:45:17 +08:00
pipeline [pipeline/chimera] reconstruct PipelineBase and Worker to support more feasible custom schedule | finish Chimera (#1595) 2022-09-19 11:44:18 +08:00
registry
tensor [autoparallel] refactored shape consistency to remove redundancy (#1591) 2022-09-13 18:30:18 +08:00
testing [NFC] polish colossalai/testing/comparison.py code style. (#1558) 2022-09-08 22:11:04 +08:00
trainer [NFC] polish ./colossalai/trainer/hooks/_lr_scheduler_hook.py code style (#1576) 2022-09-08 22:11:04 +08:00
utils [NFC] polish utils/tensor_detector/__init__.py code style (#1573) 2022-09-08 22:11:04 +08:00
zero fix move fp32 shards (#1604) 2022-09-16 17:33:16 +08:00
__init__.py [fx] provide a stable but not accurate enough version of profiler. (#1547) 2022-09-07 11:21:04 +08:00
_meta_registrations.py [fx] provide a stable but not accurate enough version of profiler. (#1547) 2022-09-07 11:21:04 +08:00
constants.py
core.py
global_variables.py
initialize.py