ColossalAI/colossalai/context
yuxuan-lou 198a74b9fd
[NFC] polish colossalai/context/random/__init__.py code style (#3327)
2023-03-30 14:19:26 +08:00
..
process_group_initializer [NFC] polish initializer_data.py code style (#3287) 2023-03-29 15:22:21 +08:00
random [NFC] polish colossalai/context/random/__init__.py code style (#3327) 2023-03-30 14:19:26 +08:00
__init__.py [polish] polish singleton and global context (#500) 2022-03-23 18:03:39 +08:00
config.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
moe_context.py [NFC] polish colossalai/context/moe_context.py code style (#2693) 2023-02-14 18:02:45 +08:00
parallel_context.py [NFC] polish colossalai/context/parallel_context.py code style (#3276) 2023-03-29 15:22:21 +08:00
parallel_mode.py updated tp layers 2022-11-02 12:19:38 +08:00
singleton_meta.py [autoparallel] refactored shape consistency to remove redundancy (#1591) 2022-09-13 18:30:18 +08:00