Elsa Granger d565df3821 [pipeline] A more general _communicate in p2p (#5062) ... * A more general _communicate * feat: finish tree_flatten version p2p * fix: update p2p api calls --------- Co-authored-by: Wenhao Chen <cwher@outlook.com>		2024-01-08 15:37:27 +08:00
..
__init__.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
_utils.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
base.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
generate.py	[inference] refactor examples and fix schedule (#5077)	2023-11-21 10:46:03 +08:00
interleaved_pp.py	[pipeline] A more general _communicate in p2p (#5062)	2024-01-08 15:37:27 +08:00
one_f_one_b.py	[pipeline] A more general _communicate in p2p (#5062)	2024-01-08 15:37:27 +08:00