Masahiro Tanaka b528f50e3d Fix buffer size for pipeline parallel and communication schedule (#2862) 1 年之前
..
__init__.py da84e60d98 add missing license info to top of all source code (#2889) 1 年之前
engine.py 98cc35b6a8 Abstract accelerator (step 3) (#2677) 1 年之前
module.py 43d58d99eb ckpt: create directories in checkpoint_engine (#2988) 1 年之前
p2p.py 98cc35b6a8 Abstract accelerator (step 3) (#2677) 1 年之前
schedule.py b528f50e3d Fix buffer size for pipeline parallel and communication schedule (#2862) 1 年之前
topology.py 316c4a43e0 Add flake8 to pre-commit checks (#2051) 2 年之前