pypots/imputation/reformer/__init__.py
"""
The package of the partially-observed time-series imputation model Reformer.
Refer to the paper
`Kitaev, Nikita, Ćukasz Kaiser, and Anselm Levskaya.
Reformer: The Efficient Transformer.
International Conference on Learning Representations, 2020.
<https://openreview.net/pdf?id=rkgNKkHtvB>`_
Notes
-----
This implementation is inspired by the official one https://github.com/google/trax/tree/master/trax/models/reformer and
https://github.com/lucidrains/reformer-pytorch
"""
# Created by Wenjie Du <wenjay.du@gmail.com>
# License: BSD-3-Clause
from .model import Reformer
__all__ = [
"Reformer",
]