Matthias Fey

PyTorch Frame: A Modular Framework for Multi-Modal Tabular Learning

2024 · arXiv (Cornell University)

We present PyTorch Frame, a PyTorch-based framework for deep learning over multi-modal tabular data. PyTorch Frame makes tabular deep learning easy by providing a PyTorch-based data structure to handle complex tabular data, introducing a model …

Group Equivariant Capsule Networks

2018 · arXiv (Cornell University)

We present group equivariant capsule networks, a framework to introduce guaranteed equivariance and invariance properties to the capsule network idea. Our work can be divided into two contributions. First, we present a generic routing by …