| # Copyright 2019 Kakao Brain | |
| # | |
| # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved. | |
| # | |
| # This source code is licensed under the BSD license found in the | |
| # LICENSE file in the root directory of this source tree. | |
| """A Pipe implementation in PyTorch.""" | |
| from .checkpoint import is_checkpointing, is_recomputing | |
| from .pipe import Pipe, WithDevice | |
| from .microbatch import NoChunk | |
| __all__ = ["Pipe", "is_checkpointing", "is_recomputing"] | |